fix: AOT compiler flags on non-sm90 (#522)

abcdabcd987 · web-flow · commit 0aa47269f9f0 · 2024-10-10T15:13:36.000-07:00
Previously non-sm90 cards incorrectly get compiled as sm90, due to the
shallow copy of the compiler flags.
diff --git a/flashinfer-aot/setup.py b/flashinfer-aot/setup.py
@@ -16,6 +16,7 @@
 
 from typing import List, Tuple
 
+import copy
 import pathlib
 import os
 import re
@@ -372,7 +373,7 @@ def __init__(self, *args, **kwargs) -> None:
             "-use_fast_math",
         ],
     }
-    extra_compile_args_sm90 = extra_compile_args.copy()
+    extra_compile_args_sm90 = copy.deepcopy(extra_compile_args)
     extra_compile_args_sm90["nvcc"].extend(
         "-gencode arch=compute_90a,code=sm_90a".split()
     )

Original file line number	Diff line number	Diff line change
`@@ -16,6 +16,7 @@`
`16`	`16`
`17`	`17`	`from typing import List, Tuple`
`18`	`18`
	`19`	`+import copy`
`19`	`20`	`import pathlib`
`20`	`21`	`import os`
`21`	`22`	`import re`
`@@ -372,7 +373,7 @@ def __init__(self, args, *kwargs) -> None:`
`372`	`373`	`"-use_fast_math",`
`373`	`374`	`],`
`374`	`375`	`}`
`375`		`- extra_compile_args_sm90 = extra_compile_args.copy()`
	`376`	`+ extra_compile_args_sm90 = copy.deepcopy(extra_compile_args)`
`376`	`377`	`extra_compile_args_sm90["nvcc"].extend(`
`377`	`378`	`"-gencode arch=compute_90a,code=sm_90a".split()`
`378`	`379`	`)`