marcsun13 HF Staff commited on
Commit
0869aae
·
verified ·
1 Parent(s): 2bbed9b

Update build/torch-universal/triton_kernels/matmul_ogs_details/opt_flags.py (#2)

Browse files

- Update build/torch-universal/triton_kernels/matmul_ogs_details/opt_flags.py (40b7a46bfb70bdc50bffbfcb008171dcfaaa58fc)

build/torch-universal/triton_kernels/matmul_ogs_details/opt_flags.py CHANGED
@@ -211,7 +211,8 @@ def make_default_opt_flags_nvidia(
211
  ns = opt_flags_nvidia.compute_num_stages(*compute_num_stages_args, ep, epilogue_effective_itemsize)
212
  if ns > num_stages:
213
  epilogue_subtile, num_stages = ep, ns
214
- assert num_stages >= 1
 
215
  if constraints.get("num_stages", None):
216
  num_stages = constraints["num_stages"]
217
 
 
211
  ns = opt_flags_nvidia.compute_num_stages(*compute_num_stages_args, ep, epilogue_effective_itemsize)
212
  if ns > num_stages:
213
  epilogue_subtile, num_stages = ep, ns
214
+ # removed due to https://huggingface.co/kernels-community/triton_kernels/discussions/1
215
+ # assert num_stages >= 1
216
  if constraints.get("num_stages", None):
217
  num_stages = constraints["num_stages"]
218