diff --git a/colossalai/kernel/cuda_native/csrc/kernels/include/context.h b/colossalai/kernel/cuda_native/csrc/kernels/include/context.h index dc80881f9..f7d75f38c 100644 --- a/colossalai/kernel/cuda_native/csrc/kernels/include/context.h +++ b/colossalai/kernel/cuda_native/csrc/kernels/include/context.h @@ -9,7 +9,7 @@ #include "cuda_util.h" class Context { -public: + public: Context() : _stream(nullptr) { CHECK_GPU_ERROR(cublasCreate(&_cublasHandle)); } @@ -30,7 +30,7 @@ public: cublasHandle_t get_cublashandle() { return _cublasHandle; } -private: + private: cudaStream_t _stream; cublasHandle_t _cublasHandle; };