mirror of
https://github.com/zebrajr/pytorch.git
synced 2025-12-07 12:21:27 +01:00
redo of https://github.com/pytorch/pytorch/pull/156136 Differential Revision: [D77327309](https://our.internmc.facebook.com/intern/diff/D77327309) I want to try land the full version first. If the ci is taking too long, we can revert back to only testing for a few names. ``` -k 'test_max_autotune_cutlass_backend_regular_mm and not test_max_autotune_cutlass_backend_regular_mm_streamk' ``` Pull Request resolved: https://github.com/pytorch/pytorch/pull/156626 Approved by: https://github.com/huydhn, https://github.com/mlazos |
||
|---|---|---|
| .. | ||
| cutlass_lib_extensions | ||
| __init__.py | ||
| cuda_cpp_scheduling.py | ||
| cuda_env.py | ||
| cuda_kernel.py | ||
| cuda_template.py | ||
| cutlass_cache.py | ||
| cutlass_presets.py | ||
| cutlass_python_evt.py | ||
| cutlass_utils.py | ||
| device_op_overrides.py | ||
| gemm_template.py | ||
| serialization.py | ||