Updates for CUTLASS 3.4.1 (#1346)

* Updates for CUTLASS 3.4.1 * minor epi change
2024-02-15 15:48:34 -05:00
parent 47a3ebbea9
commit bbe579a9e3
49 changed files with 800 additions and 451 deletions
@@ -71,6 +71,12 @@ struct GroupProblemShape {
  get_host_problem_shape(int32_t group_idx) const {
    return host_problem_shapes[group_idx];
  }
+
+  CUTLASS_HOST_DEVICE
+  bool
+  is_host_problem_shape_available() {
+    return host_problem_shapes != nullptr;
+  }
 };

 template <class ProblemShape_>
@@ -104,6 +110,12 @@ public:
  get_host_problem_shape(int32_t /* unused */ = 0) const {
    return problem_shape_;
  }
+
+  CUTLASS_HOST_DEVICE
+  bool
+  is_host_problem_shape_available() {
+    return true;
+  }
 private:
  UnderlyingProblemShape problem_shape_{};
 };