add gpu define for offlne-stream

2025-09-15 14:48:36 +08:00 · 2024-03-14 14:46:50 +08:00 · 2024-03-14 14:46:50 +08:00 · 87a8caf025
commit 87a8caf025
parent 0285a3ae5b
1 changed files with 2 additions and 0 deletions
--- a/runtime/onnxruntime/src/offline-stream.cpp
+++ b/runtime/onnxruntime/src/offline-stream.cpp
@ -36,7 +36,9 @@ OfflineStream::OfflineStream(std::map<std::string, std::string>& model_path, int
        string seg_dict_path;
    
        if(use_gpu){
+            #ifdef USE_GPU
            asr_handle = make_unique<ParaformerTorch>();
+            #endif
        }else{
            asr_handle = make_unique<Paraformer>();
        }