Skip to content

Commit d3ea4ec

Browse files
author
Zhaowu Pan
authored
Undo old cupti patch (#10367)
1 parent a14cafd commit d3ea4ec

File tree

1 file changed

+1
-3
lines changed

1 file changed

+1
-3
lines changed

ops/csrc/fp8/deep_gemm/include/deep_gemm/tma_utils.cuh

Lines changed: 1 addition & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -63,15 +63,13 @@ PFN_cuTensorMapEncodeTiled get_cuTensorMapEncodeTiled() {
6363
cudaDriverEntryPointQueryResult driver_status;
6464
void* cuTensorMapEncodeTiled_ptr = nullptr;
6565

66-
/*
6766
#if CUDA_VERSION >= 12050
6867
cudaGetDriverEntryPointByVersion("cuTensorMapEncodeTiled", &cuTensorMapEncodeTiled_ptr, 12000,
6968
cudaEnableDefault, &driver_status);
7069
#else
71-
*/
7270
cudaGetDriverEntryPoint("cuTensorMapEncodeTiled", &cuTensorMapEncodeTiled_ptr,
7371
cudaEnableDefault, &driver_status);
74-
//#endif
72+
#endif
7573

7674
if (driver_status != cudaDriverEntryPointSuccess)
7775
throw std::runtime_error("driver_status != cudaDriverEntryPointSuccess");

0 commit comments

Comments
 (0)