[LayerNorm] Compile for both sm70 and sm80

39ed597b · Tri Dao · 71f674ae · 39ed597b
Commit 39ed597b authored Nov 17, 2022 by Tri Dao
Hide whitespace changes
Inline Side-by-side

Showing with 2 additions and 2 deletions

csrc/layer_norm/setup.py csrc/layer_norm/setup.py +2 -2

No files found.
--- a/csrc/layer_norm/setup.py
+++ b/csrc/layer_norm/setup.py
@@ -98,8 +98,8 @@ if os.path.exists(os.path.join(torch_dir, "include", "ATen", "CUDAGeneratorImpl.
 raise_if_cuda_home_none("--fast_layer_norm")
 # Check, if CUDA11 is installed for compute capability 8.0
 cc_flag = []
-# cc_flag.append("-gencode")
-# cc_flag.append("arch=compute_70,code=sm_70")
+cc_flag.append("-gencode")
+cc_flag.append("arch=compute_70,code=sm_70")
 cc_flag.append("-gencode")
 cc_flag.append("arch=compute_80,code=sm_80")