From 3ae0505d99cfd7571cbad9b035a126e57321428d Mon Sep 17 00:00:00 2001
From: Jacob Hatef <jahatef@login07.frontier.olcf.ornl.gov>
Date: Sat, 24 Aug 2024 18:37:25 -0400
Subject: [PATCH] finalized instructions

---
 README.md | 13 +++++++++++--
 1 file changed, 11 insertions(+), 2 deletions(-)

diff --git a/README.md b/README.md
index 28dfa6a31..042d63732 100644
--- a/README.md
+++ b/README.md
@@ -4,10 +4,16 @@
 
 ## setup.sh 
 ```
-ROCM_VERSION=6.1 # any rocm above 6.0 will work fine
+ROCM_VERSION=6.1.3
 module reset
 module load rocm/$ROCM_VERSION
+module load ums/default  ums012/default
 module load gcc/12.2.0
+
+export ROCM_HOME=/opt/rocm-6.1.3
+export ROCM_ROOT=/opt/rocm-6.1.3
+export ROCM_PATH=/opt/rocm-6.1.3
+
 CONDA_HOME=/path/to/miniconda3
 
 export PATH=$CONDA_HOME/bin:$PATH
@@ -44,9 +50,12 @@ pip install -v --no-build-isolation --config-settings "--build-option=--cpp_ext"
 
 ## Install mamba
 ```
+git clone https://github.com/Dao-AILab/causal-conv1d.git
+cd causal-conv1d
+python setup.py install
+cd ..
 git clone https://github.com/state-spaces/mamba.git
 cd mamba
-pip install causal-conv1d>=1.4.0
 python setup.py install
 ```