fix sm75 build bug (#60583)

PaddlePaddle · Jan 5, 2024 · ed6f32d · ed6f32d
1 parent 57feb0a
commit ed6f32d
Showing 1 changed file with 10 additions and 4 deletions.
diff --git a/...phi/kernels/fusion/cutlass/cutlass_extensions/gemm/threadblock/default_dq_mma_pipelined.h b/...phi/kernels/fusion/cutlass/cutlass_extensions/gemm/threadblock/default_dq_mma_pipelined.h
@@ -78,7 +78,7 @@ template <
     /// Instruction-level tile size (concept: GemmShape)
     typename InstructionShape,
     /// Operation performed by GEMM
-    typename Operator>
+    typename Operator_>
 struct DqMma<
     ElementA,
     LayoutA,
@@ -97,9 +97,12 @@ struct DqMma<
     WarpShape,
     InstructionShape,
     2,
-    Operator,
+    Operator_,
     SharedMemoryClearOption::kNone,
     typename platform::enable_if<(ArchTag::kMinComputeCapability < 80)>::type> {
+  using OperatorInfo = arch::DetagOperator<Operator_>;
+  using Operator = typename OperatorInfo::Operator;
+
   static_assert(platform::is_same<ElementA, half_t>::value ||
                     platform::is_same<ElementA, bfloat16_t>::value,
                 "Element A must be fp16 or bf16");
@@ -227,7 +230,7 @@ template <
     /// Instruction-level tile size (concept: GemmShape)
     typename InstructionShape,
     /// Operation performed by GEMM
-    typename Operator,
+    typename Operator_,
     ///
     int RowsPerTile,
     ///
@@ -250,9 +253,12 @@ struct DqMma<
     WarpShape,
     InstructionShape,
     2,
-    Operator,
+    Operator_,
     SharedMemoryClearOption::kNone,
     typename platform::enable_if<(ArchTag::kMinComputeCapability < 80)>::type> {
+  using OperatorInfo = arch::DetagOperator<Operator_>;
+  using Operator = typename OperatorInfo::Operator;
+
   static_assert(platform::is_same<ElementA, half_t>::value ||
                     platform::is_same<ElementA, bfloat16_t>::value,
                 "Element A must be fp16 or bf16");