main
AITemplate
Repository badge with A rating
View on GitHub
Grade
Name
Complexity
Churn
Issues
B
fx2ait\fx2ait\tools\common_fx2ait.py
70 5 8
A
python\aitemplate\backend\target.py
70 1 3
A
tests\unittest\benchmark\test_group_gemm_benchmark.py
69 1 -
A
python\aitemplate\backend\cuda\attention\src\fmha_fprop_kernel_1xN.h
68 1 2
A
fx2ait\fx2ait\csrc\AITModelImpl.cpp
68 3 -
A
python\aitemplate\compiler\tensor_accessor.py
68 1 1
A
static\include\cuda_device_functions.h
65 2 1
A
python\aitemplate\utils\serialization\serdes_code.py
64 1 1
A
python\aitemplate\compiler\ops\common\elementwise.py
64 1 2
A
python\aitemplate\backend\rocm\elementwise\custom_math.h
64 1 -
A
static\include\rocm_device_functions.h
63 1 1
A
tests\unittest\compiler\test_transform_odd_alignment.py
62 1 2
A
fx2ait\fx2ait\tools\ait_subgraph_rewriter.py
61 1 3
A
python\aitemplate\backend\cuda\attention\src\fmha_block_fprop_kernel_1xN.h
61 1 2
A
python\aitemplate\backend\cuda\attention\src\fmha\gmem_tile.h
61 1 -
A
python\aitemplate\compiler\transform\memory_planning.py
60 1 -
A
tests\unittest\compiler\test_parallel_gemm_fusions.py
60 1 -
A
static\include\kernels\mem_eff_attention\gemm\custom_mma_multistage.h
60 1 1
A
python\aitemplate\compiler\transform\transform_odd_alignment.py
60 1 3
A
python\aitemplate\compiler\transform\move_view_ops.py
59 1 2