From fddc8aa126060e1ea47f679a057a1a9104fff987 Mon Sep 17 00:00:00 2001 From: RichardoLuo <14049555596@qq.com> Date: Fri, 13 May 2022 22:22:46 +0800 Subject: [PATCH] [NFC] polish colossalai/kernel/cuda_native/csrc/kernels/include/context.h code style --- colossalai/kernel/cuda_native/csrc/kernels/include/context.h | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/colossalai/kernel/cuda_native/csrc/kernels/include/context.h b/colossalai/kernel/cuda_native/csrc/kernels/include/context.h index dc80881f97c8..f7d75f38cc2b 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 @@ class Context { cublasHandle_t get_cublashandle() { return _cublasHandle; } -private: + private: cudaStream_t _stream; cublasHandle_t _cublasHandle; };