CUTLASS 3.5.1 (#1623)

* CUTLASS 3.5.1 * updates, optimizations, fixes
2024-07-29 08:46:24 -04:00
parent 56b46e2d13
commit be60a0b272
312 changed files with 19793 additions and 6775 deletions
@@ -135,14 +135,14 @@ public:
    CUTLASS_HOST_DEVICE
    Params(Layout const &layout,
           // Not needed.  Added to be compatible with strided conv epilogue.
-           conv::Conv2dProblemSize const &problem_size):
+           cutlass::Tensor4DCoord const &tensor_extent):
      Params(layout)
    { }

    CUTLASS_HOST_DEVICE
    Params(Layout const &layout,
           // Not needed.  Added to be compatible with strided conv epilogue.
-           conv::Conv3dProblemSize const &problem_size):
+           cutlass::Tensor5DCoord const &tensor_extent):
      Params(layout)
    { }

@@ -1141,7 +1141,7 @@ public:
    CUTLASS_HOST_DEVICE
    Params(Layout const &layout,
           // Not needed.  Added to be compatible with strided conv epilogue.
-           conv::Conv2dProblemSize const &problem_size):
+           cutlass::Tensor4DCoord const &tensor_extent):
      Params(layout)
    { }