OSSForks/pytorch
1
0
Fork 0
mirror of https://github.com/zebrajr/pytorch.git synced 2025-12-07 00:21:07 +01:00
Code Issues Actions 38 Packages Projects Releases Wiki Activity
1e65ccc3de
pytorch/torch/_inductor/codegen
History
Xu Han 1e65ccc3de [inductor] export kernel for gemm template. (#132580) 
Changes:
1. Move `get_export_declaration` to global scope.
2. Export kernel for gemm template.

Pull Request resolved: https://github.com/pytorch/pytorch/pull/132580
Approved by: https://github.com/ezyang
2024-08-06 18:52:22 +00:00
..
aoti_runtime [AOTI] Fix bfloat16 in CPU (#132150) 2024-08-01 22:26:30 +00:00
cuda Add basic mypy annotations to inductor (#132416) 2024-08-04 18:43:37 +00:00
rocm [ROCm][CK][Inductor] Enable addmm for CK backend to gemm max autotune (#130576) 2024-08-05 17:49:09 +00:00
xpu Flip default value for mypy disallow_untyped_defs [2/11] (#127839) 2024-06-08 18:23:08 +00:00
__init__.py
aoti_hipify_utils.py [BE][Easy][16/19] enforce style for empty lines in import segments in torch/_i*/ (#129768) 2024-07-20 16:20:58 +00:00
codegen_device_driver.py [BE][Easy][16/19] enforce style for empty lines in import segments in torch/_i*/ (#129768) 2024-07-20 16:20:58 +00:00
common.py Add None return type to init (#132335) 2024-08-01 15:26:45 +00:00
cpp_gemm_template.py [inductor] export kernel for gemm template. (#132580) 2024-08-06 18:52:22 +00:00
cpp_micro_gemm.py Add basic mypy annotations to inductor (#132416) 2024-08-04 18:43:37 +00:00
cpp_prefix.h [inductor] support vectorization for torch.argmax/min(float/int64_t)-> int64_t (#131016) 2024-08-05 04:31:53 +00:00
cpp_template_kernel.py Cpp wrapper: set args to CppWrapperKernelArgs in cpp template kernel (#129557) 2024-07-29 04:01:17 +00:00
cpp_template.py Add basic mypy annotations to inductor (#132416) 2024-08-04 18:43:37 +00:00
cpp_utils.py Revert "[inductor] export kernel for gemm template. (#132580)" 2024-08-06 13:15:15 +00:00
cpp_wrapper_cpu.py add missing profiler include in cpp code generation (#132419) 2024-08-05 13:40:47 +00:00
cpp_wrapper_cuda.py Reland "[1/2] PT2 Inductor ComboKernels - Foreach cases (#124969)" (#132182) 2024-08-05 06:57:30 +00:00
cpp.py [inductor] export kernel for gemm template. (#132580) 2024-08-06 18:52:22 +00:00
cuda_combined_scheduling.py Reland "[1/2] PT2 Inductor ComboKernels - Foreach cases (#124969)" (#132182) 2024-08-05 06:57:30 +00:00
halide.py Add basic mypy annotations to inductor (#132416) 2024-08-04 18:43:37 +00:00
memory_planning.py [BE][Easy][16/19] enforce style for empty lines in import segments in torch/_i*/ (#129768) 2024-07-20 16:20:58 +00:00
multi_kernel.py Migrate Inductor scheduler, dependencies, ir, and codegen/common to use OrderedSet (#130004) 2024-08-01 04:37:15 +00:00
simd.py Reland "[1/2] PT2 Inductor ComboKernels - Foreach cases (#124969)" (#132182) 2024-08-05 06:57:30 +00:00
triton_combo_kernel.py Reland "[1/2] PT2 Inductor ComboKernels - Foreach cases (#124969)" (#132182) 2024-08-05 06:57:30 +00:00
triton_split_scan.py Add basic mypy annotations to inductor (#132416) 2024-08-04 18:43:37 +00:00
triton_utils.py Annotate graph.py (#131400) 2024-07-23 07:04:12 +00:00
triton.py Reland "[1/2] PT2 Inductor ComboKernels - Foreach cases (#124969)" (#132182) 2024-08-05 06:57:30 +00:00
wrapper.py Reland "[1/2] PT2 Inductor ComboKernels - Foreach cases (#124969)" (#132182) 2024-08-05 06:57:30 +00:00