From 912f71d1d3454680f90ae7e99a531b807c14047a Mon Sep 17 00:00:00 2001 From: Shubham Gupta Date: Wed, 3 Apr 2024 18:03:55 -0400 Subject: [PATCH 1/5] added nvidia jetson hw support --- causal-conv1d/setup.py | 5 ++++- mamba-1p1p1/setup.py | 5 ++++- 2 files changed, 8 insertions(+), 2 deletions(-) diff --git a/causal-conv1d/setup.py b/causal-conv1d/setup.py index 12e36bf..c893b2c 100644 --- a/causal-conv1d/setup.py +++ b/causal-conv1d/setup.py @@ -104,10 +104,13 @@ def append_nvcc_threads(nvcc_extra_args): "Note: make sure nvcc has a supported version by running nvcc -V." ) - 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..ecea8ad 100644 --- a/mamba-1p1p1/setup.py +++ b/mamba-1p1p1/setup.py @@ -105,10 +105,13 @@ def append_nvcc_threads(nvcc_extra_args): "Note: make sure nvcc has a supported version by running nvcc -V." ) - 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") From a1ee456fcf3ffaafe750b676da29d214dc708938 Mon Sep 17 00:00:00 2001 From: Shubham Gupta Date: Wed, 3 Apr 2024 18:12:52 -0400 Subject: [PATCH 2/5] fix readme typo --- README.md | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/README.md b/README.md index 6ba8b1c..995bd78 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 -e causal-conv1d>=1.1.0` - `pip install -e mamba-1p1p1` From f8d2df4c8b62f55792988eaf12a10d2d3193310c Mon Sep 17 00:00:00 2001 From: Shubham Gupta Date: Wed, 3 Apr 2024 18:15:36 -0400 Subject: [PATCH 3/5] fix error with gencode --- causal-conv1d/setup.py | 1 + mamba-1p1p1/setup.py | 1 + 2 files changed, 2 insertions(+) diff --git a/causal-conv1d/setup.py b/causal-conv1d/setup.py index c893b2c..8df747d 100644 --- a/causal-conv1d/setup.py +++ b/causal-conv1d/setup.py @@ -104,6 +104,7 @@ def append_nvcc_threads(nvcc_extra_args): "Note: make sure nvcc has a supported version by running nvcc -V." ) + 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") diff --git a/mamba-1p1p1/setup.py b/mamba-1p1p1/setup.py index ecea8ad..e71dbf2 100644 --- a/mamba-1p1p1/setup.py +++ b/mamba-1p1p1/setup.py @@ -105,6 +105,7 @@ def append_nvcc_threads(nvcc_extra_args): "Note: make sure nvcc has a supported version by running nvcc -V." ) + 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") From 2a9ed331cdd297c29cfd89a511c546ea6e2fad51 Mon Sep 17 00:00:00 2001 From: Shubham Gupta Date: Wed, 3 Apr 2024 21:58:54 -0400 Subject: [PATCH 4/5] fix #33 --- README.md | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/README.md b/README.md index 995bd78..fa92d20 100644 --- a/README.md +++ b/README.md @@ -50,7 +50,7 @@ Recently the state space models (SSMs) with efficient hardware-aware designs, i. - `pip install -r vim/vim_requirements.txt` - Install ``causal-conv1d`` and ``mamba`` - - `pip install -e causal-conv1d>=1.1.0` + - `pip install -e causal-conv1d>=1.2.0` - `pip install -e mamba-1p1p1` From 13c1ac7b2b188043a5a251ab4d6e2f5992dadc4a Mon Sep 17 00:00:00 2001 From: Shubham Gupta Date: Wed, 3 Apr 2024 22:57:43 -0400 Subject: [PATCH 5/5] match causal-conv1d ver with #53 --- README.md | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/README.md b/README.md index fa92d20..4223b76 100644 --- a/README.md +++ b/README.md @@ -50,7 +50,7 @@ Recently the state space models (SSMs) with efficient hardware-aware designs, i. - `pip install -r vim/vim_requirements.txt` - Install ``causal-conv1d`` and ``mamba`` - - `pip install -e causal-conv1d>=1.2.0` + - `pip install causal-conv1d==1.1.3.post1` - `pip install -e mamba-1p1p1`