Commit 0e922da8 authored by RichardoLuo's avatar RichardoLuo Committed by binmakeswell
Browse files

[NFC] polish colossalai/kernel/cuda_native/csrc/kernels/include/context.h code style (#956)


Co-authored-by: default avatarRichardoLuo <14049555596@qq.com>
parent 8ca2a856
......@@ -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;
};
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment