restructure conv3dcoo. add 'bool cutlass' to interface in binary.cc and

binary.h.
umiswing · Sep 23, 2022 · 38f2f83 · 38f2f83
1 parent 605f11f
commit 38f2f83
Show file tree

Hide file tree

Showing 4 changed files with 5 additions and 3 deletions.
diff --git a/paddle/phi/infermeta/sparse/binary.cc b/paddle/phi/infermeta/sparse/binary.cc
@@ -60,6 +60,7 @@ void Conv3dInferMeta(const MetaTensor& x,
                      const int groups,
                      const bool subm,
                      const std::string& key,
+                     const bool cutlass,
                      MetaTensor* out,
                      MetaTensor* rulebook,
                      MetaTensor* counter) {

diff --git a/paddle/phi/infermeta/sparse/binary.h b/paddle/phi/infermeta/sparse/binary.h
@@ -30,6 +30,7 @@ void Conv3dInferMeta(const MetaTensor& x,
                      const int groups,
                      const bool subm,
                      const std::string& key,
+                     const bool cutlass,
                      MetaTensor* out,
                      MetaTensor* rulebook,
                      MetaTensor* counter);

diff --git a/paddle/phi/kernels/sparse/cpu/conv_kernel.cc b/paddle/phi/kernels/sparse/cpu/conv_kernel.cc
@@ -180,7 +180,7 @@ void Conv3dCooKernel(const Context& dev_ctx,
                      const int groups,
                      const bool subm,
                      const std::string& key,
-                     bool cutlass,
+                     const bool cutlass,
                      SparseCooTensor* out,
                      DenseTensor* rulebook,
                      DenseTensor* counter) {

diff --git a/paddle/phi/kernels/sparse/gpu/conv_kernel.cu b/paddle/phi/kernels/sparse/gpu/conv_kernel.cu
@@ -119,7 +119,7 @@ void Conv3dCooGPUKernel(const GPUContext& dev_ctx,
                         const int groups,
                         const bool subm,
                         const std::string& key,
-                        bool cutlass,
+                        const bool cutlass,
                         SparseCooTensor* out,
                         DenseTensor* rulebook,
                         DenseTensor* counter) {
@@ -396,7 +396,7 @@ void Conv3dCooKernel(const Context& dev_ctx,
                      const int groups,
                      const bool subm,
                      const std::string& key,
-                     bool cutlass,
+                     const bool cutlass,
                      SparseCooTensor* out,
                      DenseTensor* rulebook,
                      DenseTensor* counter) {