Skip to content

GitLab

Commit c1ce24db authored by flyingdown's avatar flyingdown
Browse files

enable function cuda_version/is_ffmpeg_available/is_kaldi_available/is_sox_available

parent 8bfbf473
Hide whitespace changes
Inline Side-by-side
Showing with 4 additions and 4 deletions
torchaudio/csrc/utils.cpp
View file @ c1ce24db
...@@ -45,10 +45,10 @@ c10::optional<int64_t> cuda_version() { ...@@ -45,10 +45,10 @@ c10::optional<int64_t> cuda_version() {
} // namespace } // namespace
TORCH_LIBRARY_FRAGMENT(torchaudio, m) { TORCH_LIBRARY_FRAGMENT(torchaudio, m) {
m.def("torchaudio::is_sox_available", &is_sox_available); m.def("is_sox_available", &is_sox_available);
m.def("torchaudio::is_kaldi_available", &is_kaldi_available); m.def("is_kaldi_available", &is_kaldi_available);
m.def("torchaudio::is_ffmpeg_available", &is_ffmpeg_available); m.def("is_ffmpeg_available", &is_ffmpeg_available);
m.def("torchaudio::cuda_version", &cuda_version); m.def("cuda_version", &cuda_version);
} }
} // namespace torchaudio } // namespace torchaudio
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