diff --git a/README.md b/README.md
index 6ba8b1c..4223b76 100644
--- a/README.md
+++ b/README.md
@@ -49,8 +49,8 @@ Recently the state space models (SSMs) with efficient hardware-aware designs, i.
 - Requirements: vim_requirements.txt
   - `pip install -r vim/vim_requirements.txt`
 
-- Install ``causal_conv1d`` and ``mamba``
-  - `pip install -e causal_conv1d>=1.1.0`
+- Install ``causal-conv1d`` and ``mamba``
+  - `pip install causal-conv1d==1.1.3.post1`
   - `pip install -e mamba-1p1p1`
   
   
diff --git a/causal-conv1d/setup.py b/causal-conv1d/setup.py
index 12e36bf..8df747d 100644
--- a/causal-conv1d/setup.py
+++ b/causal-conv1d/setup.py
@@ -107,7 +107,11 @@ def append_nvcc_threads(nvcc_extra_args):
     cc_flag.append("-gencode")
     cc_flag.append("arch=compute_70,code=sm_70")
     cc_flag.append("-gencode")
+    cc_flag.append("arch=compute_72,code=sm_72")
+    cc_flag.append("-gencode")
     cc_flag.append("arch=compute_80,code=sm_80")
+    cc_flag.append("-gencode")
+    cc_flag.append("arch=compute_87,code=sm_87")
     if bare_metal_version >= Version("11.8"):
         cc_flag.append("-gencode")
         cc_flag.append("arch=compute_90,code=sm_90")
diff --git a/mamba-1p1p1/setup.py b/mamba-1p1p1/setup.py
index d2a1f2f..e71dbf2 100644
--- a/mamba-1p1p1/setup.py
+++ b/mamba-1p1p1/setup.py
@@ -108,7 +108,11 @@ def append_nvcc_threads(nvcc_extra_args):
     cc_flag.append("-gencode")
     cc_flag.append("arch=compute_70,code=sm_70")
     cc_flag.append("-gencode")
+    cc_flag.append("arch=compute_72,code=sm_72")
+    cc_flag.append("-gencode")
     cc_flag.append("arch=compute_80,code=sm_80")
+    cc_flag.append("-gencode")
+    cc_flag.append("arch=compute_87,code=sm_87")
     if bare_metal_version >= Version("11.8"):
         cc_flag.append("-gencode")
         cc_flag.append("arch=compute_90,code=sm_90")