main
AITemplate
Repository badge with A rating
View on GitHub
Grade
Name
Complexity
Churn
Issues
A
python\aitemplate\backend\cuda\attention\src\fmha_fprop_fp16_kernel.sm80.cu
 1 -
A
fx2ait\fx2ait\csrc\AITModel.cpp
7 1 -
A
python\aitemplate\backend\cuda\attention\src\fmha_kernel.h
14 1 -
A
examples\04_vit\benchmark_pt.py
13 1 -
A
python\aitemplate\backend\cuda\attention\src\philox.cuh
 1 -
A
fx2ait\fx2ait\csrc\AITModelImpl.cpp
68 3 -
A
python\aitemplate\backend\cuda\b2b_bmm\grouped_classic_b2b_bmm.py
18 1 -
A
python\aitemplate\backend\cuda\b2b_bmm\classic_b2b_bmm.py
21 1 1
A
fx2ait\fx2ait\example\01_transformer_model\test_transformer_encoder.py
3 1 -
A
python\aitemplate\backend\cuda\builder_cmake.py
44 1 2
A
examples\04_vit\test_correctness.py
18 1 -
A
python\aitemplate\backend\cuda\conv2d\common_conv2d_few_channels.py
1 1 -
A
python\aitemplate\backend\cuda\conv2d\common_conv2d_bias_add_activation.py
5 1 -
A
python\aitemplate\backend\cuda\conv2d\common_conv2d_bias_activation.py
4 1 -
A
fx2ait\fx2ait\example\03_lowering_split\test_lower.py
4 1 1
A
.github\workflows\pages.yaml
 1 -
A
examples\02_detectron2\configs\mask_rcnn_R_50_FPN.yaml
 1 -
A
examples\04_vit\modeling\vision_transformer.py
41 1 -
A
fx2ait\fx2ait\csrc\AITModelImpl.h
12 2 -
A
python\aitemplate\backend\cuda\b2b_bmm\fmha_style_b2b_bmm.py
16 1 -