Skip to content

Commit e617c72

Browse files
authored
Revert windows-related changes for void *params[] in third_party/nvidia/backend/driver.py (#2939)
So it's more like what other backends have. If this was a change from an old pull request (in Triton), I guess we can roll it back. @gshimansky please take a look. Signed-off-by: Anatoly Myachev <[email protected]>
1 parent e80ef53 commit e617c72

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

third_party/nvidia/backend/driver.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -235,7 +235,7 @@ def format_of(ty):
235235
#endif
236236
237237
static void _launch(int gridX, int gridY, int gridZ, int num_warps, int num_ctas, int clusterDimX, int clusterDimY, int clusterDimZ, int shared_memory, CUstream stream, CUfunction function, CUdeviceptr global_scratch{', ' + arg_decls if len(arg_decls) > 0 else ''}) {{
238-
void *params[] = {{{', '.join(f'&arg{i}' for i in params) if params else 'NULL'}}};
238+
void *params[] = {{ {', '.join(params)} }};
239239
if (gridX*gridY*gridZ > 0) {{
240240
if (num_ctas == 1) {{
241241
CUDA_CHECK(cuLaunchKernel(function, gridX, gridY, gridZ, 32*num_warps, 1, 1, shared_memory, stream, params, 0));

0 commit comments

Comments
 (0)