Compute concurrency on HIP using Kokkos hardcoded m_maxWavesPerCU

tcclevenger · Oct 3, 2023 · 7447118 · 7447118
1 parent 41cf2e5
commit 7447118
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/core/src/HIP/Kokkos_HIP_Instance.cpp b/core/src/HIP/Kokkos_HIP_Instance.cpp
@@ -76,8 +76,8 @@ std::size_t scratch_count(const std::size_t size) {
 //----------------------------------------------------------------------------
 
 int HIPInternal::concurrency() {
-  static int const concurrency = m_deviceProp.maxThreadsPerMultiProcessor *
-                                 m_deviceProp.multiProcessorCount;
+  static int const concurrency = m_maxThreadsPerSM * m_multiProcCount;
+
   return concurrency;
 }