make only visible on device (#1071)

2023-09-07 13:00:46 -04:00
parent e01b9b5029
commit 88c0d7c726
1 changed files with 1 additions and 1 deletions
--- 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) {