lintrunner

microsoft · Mar 23, 2024 · 9d38e74 · 9d38e74
1 parent ec1eabf
commit 9d38e74
Show file tree

Hide file tree

Showing 3 changed files with 11 additions and 5 deletions.
diff --git a/onnxruntime/contrib_ops/cuda/moe/cutlass_extensions/gemm/kernel/fpA_intB_gemm.h b/onnxruntime/contrib_ops/cuda/moe/cutlass_extensions/gemm/kernel/fpA_intB_gemm.h
@@ -61,9 +61,9 @@ template <typename Mma_,                 ///! Threadblock-scoped matrix multiply
           typename Epilogue_,            ///! Epilogue
           typename ThreadblockSwizzle_,  ///! Threadblock swizzling function
           typename KernelArch,           ///! The Architecture this kernel is compiled for. Used since SIMT kernels lose
-                                         ///top-level
-                                         /// arch.
-          bool SplitKSerial              ///! If true, code supporting split-K via serial reduction is enabled.
+                                         /// top-level
+                                /// arch.
+          bool SplitKSerial  ///! If true, code supporting split-K via serial reduction is enabled.
           >
 struct GemmFpAIntB {
   using Mma = Mma_;

diff --git a/onnxruntime/contrib_ops/cuda/moe/cutlass_extensions/gemm_configs.h b/onnxruntime/contrib_ops/cuda/moe/cutlass_extensions/gemm_configs.h
@@ -93,7 +93,10 @@ enum class EpilogueScheduleType {
         // architectures older than hopper, the epilogue is always performed by the same thread block as the main loop.
 };
 
-enum class ClusterShape { ClusterShape_1x1x1, ClusterShape_2x1x1, ClusterShape_1x2x1, ClusterShape_2x2x1 };
+enum class ClusterShape { ClusterShape_1x1x1,
+                          ClusterShape_2x1x1,
+                          ClusterShape_1x2x1,
+                          ClusterShape_2x2x1 };
 
 struct CutlassGemmConfig {
   CutlassTileConfig tile_config = CutlassTileConfig::ChooseWithHeuristic;

diff --git a/onnxruntime/contrib_ops/cuda/moe/cutlass_extensions/weight_only_quant_op.h b/onnxruntime/contrib_ops/cuda/moe/cutlass_extensions/weight_only_quant_op.h
@@ -36,7 +36,10 @@
 
 namespace cutlass {
 
-enum class WeightOnlyQuantOp { UNDEFINED, PER_COLUMN_SCALE_ONLY, FINEGRAINED_SCALE_ONLY, FINEGRAINED_SCALE_AND_ZEROS };
+enum class WeightOnlyQuantOp { UNDEFINED,
+                               PER_COLUMN_SCALE_ONLY,
+                               FINEGRAINED_SCALE_ONLY,
+                               FINEGRAINED_SCALE_AND_ZEROS };
 
 constexpr bool isFinegrained(WeightOnlyQuantOp op) {
   return op == WeightOnlyQuantOp::FINEGRAINED_SCALE_AND_ZEROS || op == WeightOnlyQuantOp::FINEGRAINED_SCALE_ONLY;