Aaron Orenstein 6503b4a96e Update to using mypy 1.15 (#154054) ... The BC break isn't real - mypy decided to start complaining about the way we were typing that function. Pull Request resolved: https://github.com/pytorch/pytorch/pull/154054 Approved by: https://github.com/Skylion007		2025-05-24 04:30:57 +00:00
..
cutlass_lib_extensions	[Cutlass] E2E Tests for EVT (#152815)	2025-05-17 12:29:10 +00:00
__init__.py
cuda_cpp_scheduling.py	[Cutlass] Enable fusion with FusedSchedulerNodes (#153588)	2025-05-17 12:29:10 +00:00
cuda_env.py	Revert "[BE]: Enable RUFF TRY400 rule - log.exception (#153473)"	2025-05-16 08:29:26 +00:00
cuda_kernel.py	[inductor][cutlass backend] Add 2 stage autotuning aka prescreening (#153335)	2025-05-23 17:12:25 +00:00
cuda_template.py	[Cutlass] Support float8_e4m3fn GEMM (#153890)	2025-05-22 08:37:33 +00:00
cutlass_presets.py	[pytorch] reland of [cutlass backend] delay construction of cutlass presets to when called (#151875) (#152031)	2025-04-24 05:36:36 +00:00
cutlass_python_evt.py	[Cutlass] Support float8_e4m3fn GEMM (#153890)	2025-05-22 08:37:33 +00:00
cutlass_utils.py	[Cutlass] Support float8_e4m3fn GEMM (#153890)	2025-05-22 08:37:33 +00:00
device_op_overrides.py	[AOTI] Embed cubin files into .so (#150739)	2025-05-19 01:11:46 +00:00
gemm_template.py	Update to using mypy 1.15 (#154054)	2025-05-24 04:30:57 +00:00
serialization.py	[cutlass backend] Add serializer for cutlass ops (#153894)	2025-05-21 22:01:40 +00:00