Skip to content

Commit

Permalink
[runtime/ipex] Add ITN support in ipex path (#2246)
Browse files Browse the repository at this point in the history
* update ITN dependency

* upgrade torch version in README.md
  • Loading branch information
ZailiWang committed Dec 15, 2023
1 parent 3d73821 commit 907e986
Show file tree
Hide file tree
Showing 2 changed files with 7 additions and 2 deletions.
4 changes: 4 additions & 0 deletions runtime/ipex/CMakeLists.txt
Original file line number Diff line number Diff line change
Expand Up @@ -10,6 +10,7 @@ option(GRPC "whether to build with gRPC" OFF)
option(WEBSOCKET "whether to build with websocket" OFF)
option(HTTP "whether to build with http" OFF)
option(TORCH "whether to build with Torch" ON)
option(ITN "whether to use WeTextProcessing" ON)
option(IPEX "whether to build with Torch+IPEX" ON)

set(CMAKE_VERBOSE_MAKEFILE OFF)
Expand All @@ -30,6 +31,9 @@ include_directories(
${CMAKE_CURRENT_SOURCE_DIR}
${CMAKE_CURRENT_SOURCE_DIR}/kaldi
)
if(ITN)
include(wetextprocessing)
endif()

# Build all libraries
add_subdirectory(utils)
Expand Down
5 changes: 3 additions & 2 deletions runtime/ipex/README.md
Original file line number Diff line number Diff line change
Expand Up @@ -62,9 +62,10 @@ git clone https://github.com/wenet-e2e/wenet
cd wenet
pip install -r requirements.txt
```
Installation of IPEX
Upgrading of PyTorch and TorchAudio, followed by the installation of IPEX
``` sh
pip install intel_extension_for_pytorch==2.0.100 -f https://developer.intel.com/ipex-whl-stable-cpu
pip install torch==2.0.1 torchaudio==2.0.2 --index-url https://download.pytorch.org/whl/cpu --force-reinstall
pip install intel_extension_for_pytorch==2.0.100
```

Installation of related tools: Intel® OpenMP and TCMalloc
Expand Down

0 comments on commit 907e986

Please sign in to comment.