squall/cutlass
1
0
Fork 0
Code Issues Pull Requests Actions 1 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
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 deletions
Show Stats 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] // Returns the linearized index of the output tile corresponding to the tile with offset [L, M, K]
CUTLASS_HOST_DEVICE CUTLASS_DEVICE
static int static int
output_tile_index(Params const& params, WorkTileInfo const& work_tile_info) { output_tile_index(Params const& params, WorkTileInfo const& work_tile_info) {
if (params.splits_ > 1) { if (params.splits_ > 1) {