mirror of
https://github.com/zebrajr/pytorch.git
synced 2025-12-07 00:21:07 +01:00
Differential Revision: [D77186982](https://our.internmc.facebook.com/intern/diff/D77186982/) source: https://pypi.org/project/nvidia-cutlass/ If users want to use it, they can install pytorch through wheel, git clone cutlass, and specify cutlass path via TORCHINDUCTOR_CUTLASS_DIR Pull Request resolved: https://github.com/pytorch/pytorch/pull/156651 Approved by: 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 | ||