Skip to content

GitLab

Commit f456860f authored by lishen's avatar lishen
Browse files

warpctc for dcu

parent ffdb193b
Show whitespace changes
Inline Side-by-side
Showing with 4 additions and 0 deletions
README_HIP.md
View file @ f456860f
...@@ -33,12 +33,14 @@ ...@@ -33,12 +33,14 @@
- 使用python安装 - 使用python安装
```shell ```shell
cd pytorch_binding
python setup.py install python setup.py install
``` ```
- 使用python编译whl包 - 使用python编译whl包
```shell ```shell
cd pytorch_binding
python setup.py bdist_wheel python setup.py bdist_wheel
``` ```
...@@ -47,12 +49,14 @@ python setup.py bdist_wheel ...@@ -47,12 +49,14 @@ python setup.py bdist_wheel
- 验证warpctc的loss正确性（CPU和GPU的一致性） - 验证warpctc的loss正确性（CPU和GPU的一致性）
```shell ```shell
cd pytorch_binding/tests
python3 test_gpu.py python3 test_gpu.py
``` ```
- 验证warpctc的loss的GPU加速效果 - 验证warpctc的loss的GPU加速效果
```shell ```shell
cd pytorch_binding/tests
python3 test_gpu_speed.py python3 test_gpu_speed.py
``` ```
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment