Set the number of teams as the loop bound.

rcclay · May 4, 2024 · 89091d6 · 89091d6
1 parent efdf45b
commit 89091d6
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/Platforms/OMPTarget/ompBLAS.cpp b/src/Platforms/OMPTarget/ompBLAS.cpp
@@ -198,7 +198,7 @@ ompBLAS_status gemv_impl(ompBLAS_handle& handle,
     if (incx != 1 || incy != 1)
       throw std::runtime_error("incx !=1 or incy != 1 are not implemented in ompBLAS::gemv_impl!");
 
-    PRAGMA_OFFLOAD("omp target teams distribute num_teams(n) is_device_ptr(A, x, y)")
+    PRAGMA_OFFLOAD("omp target teams distribute num_teams(m) is_device_ptr(A, x, y)")
     for (uint32_t i = 0; i < m; i++)
     {
       T dot_sum(0);