NVIDIA/cutlass
1
0
Fork 0
mirror of https://github.com/NVIDIA/cutlass.git synced 2026-05-11 17:00:05 +00:00
Code Issues Packages Projects Releases Wiki Activity

make only visible on device (#1071)

Browse Source
This commit is contained in:
Driss Guessous
2023-09-07 13:00:46 -04:00
committed by GitHub
parent e01b9b5029
commit 88c0d7c726
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
include/cutlass/gemm/kernel/sm90_tile_scheduler_stream_k.hpp
View File

@@ -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) {