Skip to content

Commit fdda958

Browse files
authored
bugfix: fix the signature of CutlassSegmentGEMMSM90 (#827)
Follow up of #823 , the `CutlassSegmentGEMMSM90` API do not have member `plan_info_vec`
1 parent 39323cd commit fdda958

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

csrc/flashinfer_gemm_sm90_ops.cu

+1-1
Original file line numberDiff line numberDiff line change
@@ -19,7 +19,7 @@ void CutlassSegmentGEMMSM90(at::Tensor float_workspace_buffer, at::Tensor int_wo
1919
at::Tensor all_problems, at::Tensor x_ptr, at::Tensor w_ptr,
2020
at::Tensor y_ptr, at::Tensor x_stride, at::Tensor weight_stride,
2121
at::Tensor y_stride, at::Tensor empty_x_data, bool weight_column_major,
22-
at::Tensor plan_info_vec, int64_t cuda_stream);
22+
int64_t cuda_stream);
2323

2424
TORCH_LIBRARY_FRAGMENT(TORCH_EXTENSION_NAME, m) {
2525
// "Cutlass Segment GEMM operator for SM90"

0 commit comments

Comments
 (0)