Skip to content

Commit

Permalink
Update table6 README.
Browse files Browse the repository at this point in the history
  • Loading branch information
lcy-seso committed Sep 18, 2024
1 parent 2074b3b commit fc061ce
Showing 1 changed file with 12 additions and 2 deletions.
14 changes: 12 additions & 2 deletions artifacts/table6/README.md
Original file line number Diff line number Diff line change
Expand Up @@ -52,7 +52,7 @@ We have prepared a testing environment on the provided server to run the tests.
./run_all_ncu_cutlass.sh
```

2. The script [run_all_ncu_flash2.sh](../run_all_ncu_pt.sh) is used to run the test for Flash Attention 2, implemented in PyTorch.
2. The script [run_all_ncu_flash2.sh](../run_all_ncu_flash2.sh) is used to run the test for Flash Attention 2, from the author's official implementation.
```bash
sudo -i # Switch to root account
Expand All @@ -67,7 +67,7 @@ We have prepared a testing environment on the provided server to run the tests.
```bash
sudo -i # Switch to root account
cd /home/sosp/nnfusion/artifacts
./run_all_ncu_pt.sh
./run_all_ncu_ft.sh
```
4. The script [run_all_ncu_pt.sh](../run_all_ncu_pt.sh) is used to run the test for BigBird, implemented in PyTorch.
Expand All @@ -78,4 +78,14 @@ We have prepared a testing environment on the provided server to run the tests.
# Choose the environment you want to test
source /home/sosp/env/torch_env.sh
./run_all_ncu_pt.sh
```
5. The script [run_all_ncu_triton.sh](../run_all_ncu_triton.sh) is used to run the test for BigBird and Flash Attention, implemented in Triton.
```bash
sudo -i # Switch to root account
cd /home/sosp/nnfusion/artifacts
# Choose the environment you want to test
source /home/sosp/env/torch_env.sh
./run_all_ncu_triton.sh
```

0 comments on commit fc061ce

Please sign in to comment.