make only visible on device (#1071)

SNU-ARC · Feb 7, 2024 · 95bc8ef · 95bc8ef
1 parent 4af3759
commit 95bc8ef
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/include/cutlass/gemm/kernel/sm90_tile_scheduler_stream_k.hpp b/include/cutlass/gemm/kernel/sm90_tile_scheduler_stream_k.hpp
@@ -556,7 +556,7 @@ class PersistentTileSchedulerSm90StreamK {
   }
 
   // Returns the linearized index of the output tile corresponding to the tile with offset [L, M, K]
-  CUTLASS_HOST_DEVICE
+  CUTLASS_DEVICE
   static int
   output_tile_index(Params const& params, WorkTileInfo const& work_tile_info) {
     if (params.splits_ > 1) {