send dyn shmem param to hip runtime

masahi · Jun 28, 2021 · aa4b90d · aa4b90d
1 parent 86c3c34
commit aa4b90d
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/src/runtime/rocm/rocm_module.cc b/src/runtime/rocm/rocm_module.cc
@@ -167,10 +167,11 @@ class ROCMWrappedFunc {
     ThreadWorkLoad wl = thread_axis_cfg_.Extract(args);
     void* config[] = {HIP_LAUNCH_PARAM_BUFFER_POINTER, packed_args, HIP_LAUNCH_PARAM_BUFFER_SIZE,
                       &packed_nbytes, HIP_LAUNCH_PARAM_END};
+    LOG(INFO) << "dynamic shared mem size: " << wl.dyn_shmem_size;
     // HIP supports only extra_args.
     ROCM_DRIVER_CALL(hipModuleLaunchKernel(
         fcache_[device_id], wl.grid_dim(0), wl.grid_dim(1), wl.grid_dim(2), wl.block_dim(0),
-        wl.block_dim(1), wl.block_dim(2), 0, strm, nullptr, reinterpret_cast<void**>(&config)));
+        wl.block_dim(1), wl.block_dim(2), wl.dyn_shmem_size, strm, nullptr, reinterpret_cast<void**>(&config)));
   }
 
  private: