pytorch/torch/_inductor/codegen
2024-01-23 01:16:15 +00:00
..
aoti_runtime [AOTInductor] Allow user to explicitly specify Device to run on (#117413) 2024-01-17 19:28:04 +00:00
cuda [BE]: Enable F821 and fix bugs (#116579) 2024-01-01 08:40:46 +00:00
__init__.py
common.py Rename unbacked SymInt prefix to u (#117859) 2024-01-22 20:53:47 +00:00
cpp_prefix.h enable fp8 cast for inductor CPU (#117737) 2024-01-23 01:16:15 +00:00
cpp.py enable fp8 cast for inductor CPU (#117737) 2024-01-23 01:16:15 +00:00
cuda_combined_scheduling.py [Inductor CUTLASS backend] Epilogue fusion codegen (Step 1) (#110890) 2023-11-06 19:42:10 +00:00
memory_planning.py Properly type CachedFunction & rename to CachedMethod (#114161) 2023-11-25 01:30:23 +00:00
multi_kernel.py [inductor] multi-kernel support (#103469) 2024-01-18 23:16:31 +00:00
triton_foreach.py [inductor] make inductor work with new triton compile interface (#115878) 2023-12-22 00:09:29 +00:00
triton_utils.py additional support for float8_e4m3fnuz and _e5m2fnuz (#115214) 2024-01-22 18:33:41 +00:00
triton.py Rename unbacked SymInt prefix to u (#117859) 2024-01-22 20:53:47 +00:00
wrapper.py [cpp_wrapper] Change CppWrapperCodeCache to use faster python binding (#117693) 2024-01-21 16:07:52 +00:00