|
Go up
|
— |
— |
|
|
__pycache__/
|
— |
|
|
|
aoti_runtime/
|
— |
|
|
|
cuda/
|
— |
|
|
|
cutedsl/
|
— |
|
|
|
cutlass/
|
— |
|
|
|
mtia/
|
— |
|
|
|
nv_universal_gemm/
|
— |
|
|
|
rocm/
|
— |
|
|
|
xpu/
|
— |
|
|
|
__init__.py
|
0 B |
|
|
|
aoti_hipify_utils.py
|
1.4 KiB |
|
|
|
block_analysis.py
|
8.0 KiB |
|
|
|
common.py
|
105 KiB |
|
|
|
cpp.py
|
234 KiB |
|
|
|
cpp_bmm_template.py
|
9.4 KiB |
|
|
|
cpp_flex_attention_template.py
|
40 KiB |
|
|
|
cpp_gemm_template.py
|
77 KiB |
|
|
|
cpp_grouped_gemm_template.py
|
21 KiB |
|
|
|
cpp_micro_gemm.py
|
76 KiB |
|
|
|
cpp_template.py
|
4.9 KiB |
|
|
|
cpp_template_kernel.py
|
25 KiB |
|
|
|
cpp_utils.py
|
28 KiB |
|
|
|
cpp_wrapper_cpu.py
|
132 KiB |
|
|
|
cpp_wrapper_cpu_array_ref.py
|
38 KiB |
|
|
|
cpp_wrapper_gpu.py
|
54 KiB |
|
|
|
cpp_wrapper_mps.py
|
12 KiB |
|
|
|
cpu_device_op_overrides.py
|
694 B |
|
|
|
cuda_combined_scheduling.py
|
7.3 KiB |
|
|
|
custom_extern_kernel_codegen.py
|
3.0 KiB |
|
|
|
debug_utils.py
|
11 KiB |
|
|
|
halide.py
|
65 KiB |
|
|
|
memory_planning.py
|
26 KiB |
|
|
|
mps.py
|
47 KiB |
|
|
|
mps_device_op_overrides.py
|
671 B |
|
|
|
multi_kernel.py
|
23 KiB |
|
|
|
pallas.py
|
196 KiB |
|
|
|
python_wrapper_mtia.py
|
988 B |
|
|
|
segmented_tree.py
|
8.0 KiB |
|
|
|
simd.py
|
124 KiB |
|
|
|
simd_kernel_features.py
|
24 KiB |
|
|
|
subgraph.py
|
24 KiB |
|
|
|
triton.py
|
272 KiB |
|
|
|
triton_combo_kernel.py
|
50 KiB |
|
|
|
triton_split_scan.py
|
7.5 KiB |
|
|
|
triton_utils.py
|
11 KiB |
|
|
|
wrapper.py
|
170 KiB |
|
|
|
wrapper_fxir.py
|
45 KiB |
|
|