update rocm build instruction

microsoft · Apr 25, 2023 · 6485a19 · 6485a19
1 parent f9a6c63
commit 6485a19
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/docs/build/eps.md b/docs/build/eps.md
@@ -663,13 +663,13 @@ See more information on the ROCm Execution Provider [here](../execution-provider
 ### Build Instructions
 {: .no_toc }
 
-#### Build for inference (Linux)
+#### Linux
 
 ```bash
 ./build.sh --config <Release|Debug|RelWithDebInfo> --use_rocm --rocm_home <path to ROCm home> --build_wheel --update --build --parallel
 ```
 
-#### Build for inference (Linux, with Composable Kernel)
+#### Linux (with Composable Kernel)
 ```bash
 ./build.sh --config <Release|Debug|RelWithDebInfo> --use_rocm --rocm_home <path to ROCm home> --build_wheel --update --build --parallel --cmake_extra_defines onnxruntime_USE_COMPOSABLE_KERNEL=ON
 ```

diff --git a/docs/build/training.md b/docs/build/training.md
@@ -94,9 +94,9 @@ The default AMD GPU build requires ROCm software toolkit installed on the system
 2. Create the ONNX Runtime wheel
 
    * Change to the ONNX Runtime repo base folder: `cd onnxruntime`
-   * Run `./build.sh --config RelWithDebInfo --enable_training --build_wheel --use_rocm --rocm_home /opt/rocm --nccl_home /opt/rocm --mpi_home <location for openmpi>`
+   * Run `./build.sh --config <Release|Debug|RelWithDebInfo> --enable_training --use_rocm --rocm_home /opt/rocm --nccl_home /opt/rocm --mpi_home <location for openmpi> --build_wheel --update --build --parallel`
 
-    This produces the .whl file in `./build/Linux/RelWithDebInfo/dist` for ONNX Runtime Training.
+    This produces the .whl file in `./build/Linux/<Release|Debug|RelWithDebInfo>/dist` for ONNX Runtime Training.
 
 ## DNNL and MKLML