Skip to content

GitLab

Unverified Commit 76314a4b authored by Caroline Chen's avatar Caroline Chen Committed by GitHub
Browse files

Expose USE_CUDA in build (#1609)

parent 4b61ae3f
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
build_tools/setup_helpers/extension.py
View file @ 76314a4b
...@@ -38,7 +38,7 @@ _BUILD_SOX = False if platform.system() == 'Windows' else _get_build("BUILD_SOX" ...@@ -38,7 +38,7 @@ _BUILD_SOX = False if platform.system() == 'Windows' else _get_build("BUILD_SOX"
_BUILD_KALDI = False if platform.system() == 'Windows' else _get_build("BUILD_KALDI", True) _BUILD_KALDI = False if platform.system() == 'Windows' else _get_build("BUILD_KALDI", True)
_BUILD_TRANSDUCER = _get_build("BUILD_TRANSDUCER") _BUILD_TRANSDUCER = _get_build("BUILD_TRANSDUCER")
_USE_ROCM = _get_build("USE_ROCM") _USE_ROCM = _get_build("USE_ROCM")
_USE_CUDA = torch.cuda.is_available() _USE_CUDA = _get_build("USE_CUDA", torch.cuda.is_available())
def get_ext_modules(): def get_ext_modules():
......
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