8000 Update on "[Cutlass] Enable fusion with FusedSchedulerNodes" · pytorch/pytorch@6685d54 · GitHub
[go: up one dir, main page]

Skip to content

Commit 6685d54

Browse files
committed
Update on "[Cutlass] Enable fusion with FusedSchedulerNodes"
cc voznesenskym penguinwu EikanWang jgong5 Guobing-Chen XiaobingSuper zhuhaozhe blzheng wenzhe-nrv jiayisunx ipiszy chenyang78 kadeng muchulee8 amjames chauhang aakhundov [ghstack-poisoned]
2 parents 618a193 + a8ff16d commit 6685d54

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

torch/_inductor/config.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1297,6 +1297,8 @@ class aot_inductor:
12971297

12981298

12991299
class cuda:
1300+
"""Settings for cuda backend, today this consists of cutlass"""
1301+
13001302
# CUDA arch to use for CUDA template kernel compilation.
13011303
# e.g. "70", "75", "80", "90", etc.
13021304
# When arch is None, Inductor uses torch.cuda.get_device_capability(0).

0 commit comments

Comments
 (0)
0