                    const char *backendStr_, const char *dtypeStr_)
      : aDims{m, n}, cDims{m, n}, numLayers_(numLayers_),
        asyncLaunchSize_(asyncLaunchSize_), numCores_(numCores_),
        backendStr_(backendStr_), dtypeStr_(dtypeStr_) {}

  void setup() override {

    // Setup host manager
    std::vector<std::unique_ptr<runtime::DeviceConfig>> configs;