Skip to content

Commit f24b4dd

Browse files
committed
Fix:
1 parent 4398043 commit f24b4dd

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

csrc/generation/tune_cublaslt_gemm.cu

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -575,7 +575,7 @@ void GEMMInt8<int8_t, int32_t, CUBLASLTContext>(const CUBLASLTContext& dev_ctx,
575575
if (is_test) {
576576
std::vector<algoSelect_t> algos;
577577
// Select //
578-
FindAlgo(dev_ctx.handle_,
578+
FindAlgo(dev_ctx.handle,
579579
m,
580580
n,
581581
k,
@@ -644,7 +644,7 @@ void GEMMInt8<int8_t, int32_t, CUBLASLTContext>(const CUBLASLTContext& dev_ctx,
644644
paddle::DataType::UINT8,
645645
paddle::GPUPlace());
646646
void* workspace_ptr = workspace.data<uint8_t>();
647-
CUDA_CHECK(cublasLtMatmulAlgoInit(dev_ctx.handle_,
647+
CUDA_CHECK(cublasLtMatmulAlgoInit(dev_ctx.handle,
648648
cudaComputeType,
649649
CUDA_R_32I,
650650
CUDA_R_8I,
@@ -678,7 +678,7 @@ void GEMMInt8<int8_t, int32_t, CUBLASLTContext>(const CUBLASLTContext& dev_ctx,
678678
auto start = std::chrono::high_resolution_clock::now();
679679
const int repeats = 10;
680680
for (int loop = 0; loop < repeats; loop++) {
681-
CUDA_CHECK(cublasLtMatmul(dev_ctx.handle_,
681+
CUDA_CHECK(cublasLtMatmul(dev_ctx.handle,
682682
matmul_desc_,
683683
&alpha_,
684684
B_dev,

0 commit comments

Comments
 (0)