floppyrj45/lingbot-map
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

update gct

Browse Source
This commit is contained in:
LinZhuoChen
2026-04-21 22:11:44 +08:00
parent 783ea8c3dd
commit 59042da3e5
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
gct_profile.py
View File

@@ -224,7 +224,7 @@ def main():
'at a small accuracy cost).') 'at a small accuracy cost).')
parser.add_argument('--backend', choices=['sdpa', 'flashinfer', 'both'], default='flashinfer') parser.add_argument('--backend', choices=['sdpa', 'flashinfer', 'both'], default='flashinfer')
parser.add_argument('--dtype', choices=['bf16', 'fp32', 'both'], default='bf16') parser.add_argument('--dtype', choices=['bf16', 'fp32', 'both'], default='bf16')
parser.add_argument('--compile', action='store_true', default=False, parser.add_argument('--compile', action='store_true', default=True,
help='torch.compile hot modules (reduce-overhead) and drop point_head. ' help='torch.compile hot modules (reduce-overhead) and drop point_head. '
'Typically ~5 FPS faster at 518×378.') 'Typically ~5 FPS faster at 518×378.')
parser.add_argument('--fa3', action='store_true', parser.add_argument('--fa3', action='store_true',