lint c++

xadupre · Sep 1, 2023 · db31126 · db31126
1 parent 55b06ea
commit db31126
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 2 deletions.
diff --git a/onnxruntime/contrib_ops/cuda/math/gemm_float8.cc b/onnxruntime/contrib_ops/cuda/math/gemm_float8.cc
@@ -7,6 +7,7 @@
 #include "core/providers/cuda/shared_inc/fpgeneric.h"
 #include "core/providers/cpu/math/gemm_helper.h"
 #include "gemm_float8.h"
+#include <string>
 
 using namespace ONNX_NAMESPACE;
 

diff --git a/onnxruntime/contrib_ops/cuda/math/gemm_float8.cu b/onnxruntime/contrib_ops/cuda/math/gemm_float8.cu
@@ -5,6 +5,7 @@
 #include "core/providers/cuda/cu_inc/common.cuh"
 #include "core/providers/cuda/shared_inc/cuda_utils.h"
 #include <algorithm>
+#include <utility>
 #include <cuda_runtime.h>
 
 namespace onnxruntime {
@@ -379,7 +380,7 @@ Status GemmFloat8::ComputeGemm(
   // The workspace should be allocated once from OpKernelContext assuming
   // only one cuda function is running at a time (which is not necessarily true
   // with H100).
-  size_t workspaceSize = (size_t)(1 << 25);  // suggested fixed value 32Mb
+  size_t workspaceSize = static_cast<size_t>(1 << 25);  // suggested fixed value 32Mb
   cublasLtMatmulPreference_t preference = nullptr;
   cublasLtMatmulPreferenceCreate(&preference);
   cublasLtMatmulPreferenceSetAttribute(preference,

diff --git a/onnxruntime/contrib_ops/cuda/math/gemm_float8.h b/onnxruntime/contrib_ops/cuda/math/gemm_float8.h
@@ -60,7 +60,7 @@ class GemmFloat8 final : public onnxruntime::cuda::CudaKernel {
   cublasComputeType_t compute_type_;
   cublasLtEpilogue_t epilogue_;
 
-  // TODO: add epilogue (= activation function, Relu or Gelu are available).
+  // TODO(xadupre): add epilogue (= activation function, Relu or Gelu are available).
 };
 
 }  // namespace cuda