Skip to content

Commit 0481988

Browse files
authored
clean some CUDA_VERSION >= 11020 (PaddlePaddle#75865)
1 parent 45f3410 commit 0481988

File tree

2 files changed

+4
-4
lines changed

2 files changed

+4
-4
lines changed

paddle/phi/kernels/funcs/cublaslt.h

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -104,7 +104,7 @@ class CublasLtHelper {
104104
"refer https://docs.nvidia.com/cuda/cublas/index.html to get more "
105105
"information"));
106106

107-
#if CUDA_VERSION >= 11020
107+
#if defined(PADDLE_WITH_CUDA)
108108

109109
int algoId = 21;
110110
int swizzle = 0;
@@ -189,7 +189,7 @@ class CublasLtHelper {
189189
C_desc_,
190190
C_dev,
191191
C_desc_,
192-
#if CUDA_VERSION >= 11020
192+
#if defined(PADDLE_WITH_CUDA)
193193
&algo_,
194194
workspace,
195195
workspace_size_,

paddle/phi/kernels/gpu/llm_int8_linear_kernel.cu

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -19,7 +19,7 @@
1919
#include "paddle/phi/kernels/elementwise_add_kernel.h"
2020
#include "paddle/phi/kernels/funcs/broadcast_function.h"
2121
#include "paddle/phi/kernels/funcs/elementwise_functor.h"
22-
#if defined(PADDLE_WITH_CUDA) && CUDA_VERSION >= 11020
22+
#if defined(PADDLE_WITH_CUDA)
2323
#include "paddle/phi/kernels/impl/llm_int8_matmul_kernel_impl.h"
2424
#endif
2525

@@ -33,7 +33,7 @@ void llm_int8_compute(const Context& dev_ctx,
3333
const DenseTensor& weight_scale,
3434
const float threshold,
3535
DenseTensor* out) {
36-
#if defined(PADDLE_WITH_CUDA) && CUDA_VERSION >= 11020
36+
#if defined(PADDLE_WITH_CUDA)
3737
DenseTensor cublaslt_workspace;
3838
cublaslt_workspace.Resize({{3000000}});
3939
dev_ctx.template Alloc<int8_t>(&cublaslt_workspace);

0 commit comments

Comments
 (0)