From 88c0d7c72637b8f495ee28b58ecddfd87b6ca2ab Mon Sep 17 00:00:00 2001
From: Driss Guessous <32754868+drisspg@users.noreply.github.com>
Date: Thu, 7 Sep 2023 13:00:46 -0400
Subject: [PATCH] make only visible on device (#1071)

---
 include/cutlass/gemm/kernel/sm90_tile_scheduler_stream_k.hpp | 2 +-
 1 file changed, 1 insertion(+), 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
index ff366c32..91243544 100644
--- 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 @@ public:
   }
 
   // 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) {