Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Onnxrt execution provider configs #992

Merged
merged 34 commits into from
Jul 5, 2024
Merged
Show file tree
Hide file tree
Changes from 1 commit
Commits
Show all changes
34 commits
Select commit Hold shift + click to select a range
3c1879d
onnxrt ep configs
manickavela29 Jun 6, 2024
e69d495
updating config
manickavela29 Jun 24, 2024
47cdf99
updating python api
manickavela29 Jun 24, 2024
e29ff24
attempting fixing pybind
manickavela29 Jun 24, 2024
d139c1d
patch for pybind and clean
manickavela29 Jun 24, 2024
79f45ad
clean up
manickavela-uni Jun 25, 2024
207f4a1
python api complete
manickavela29 Jun 25, 2024
df2e2e1
Apply suggestions from code review
manickavela29 Jun 26, 2024
c477339
fix and update
manickavela29 Jun 26, 2024
f20157c
lint fix
manickavela29 Jun 26, 2024
22d73e5
device_id fix
manickavela29 Jun 26, 2024
e63dc19
tidy-clang and lint
manickavela29 Jun 26, 2024
021dc01
Update sherpa-onnx/python/csrc/provider-config.cc
manickavela29 Jun 26, 2024
9c6be4f
updating python api cmake
manickavela29 Jun 26, 2024
0022ae4
pybind fix
manickavela29 Jun 26, 2024
b4f1985
fix int32_t
manickavela29 Jun 26, 2024
4011948
uint32_t back and jni fix
manickavela29 Jun 26, 2024
1cfd469
JNI and python fix
manickavela29 Jun 26, 2024
2ef7a7c
Update sherpa-onnx/csrc/provider-config.cc
manickavela29 Jun 26, 2024
fa001b7
removing from offline
manickavela29 Jun 26, 2024
be3aa27
Apply suggestions from code review
manickavela29 Jun 27, 2024
e2dc60c
bug fix
manickavela29 Jun 27, 2024
4f5a58f
handling uint and attempting python fix
manickavela-uni Jun 28, 2024
389cbf5
pybind attempt
manickavela29 Jun 28, 2024
bbfea38
python-dump
manickavela29 Jun 29, 2024
2f188bb
python interface
manickavela29 Jul 3, 2024
dc8bfb2
lint fix
manickavela29 Jul 3, 2024
0982ee7
Update sherpa-onnx/python/sherpa_onnx/online_recognizer.py
manickavela29 Jul 4, 2024
4e2ede0
clean up
manickavela29 Jul 4, 2024
6ab0567
fixing keyword spotter
manickavela29 Jul 4, 2024
04a9d8d
keyword device
manickavela29 Jul 4, 2024
1a84eaf
clean up
manickavela29 Jul 4, 2024
dfe9a19
update condition and fix
manickavela-uni Jul 4, 2024
4cc82fc
Update provider-config.cc
manickavela29 Jul 5, 2024
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
8 changes: 5 additions & 3 deletions sherpa-onnx/csrc/provider-config.cc
Original file line number Diff line number Diff line change
Expand Up @@ -113,9 +113,11 @@ void ProviderConfig::Register(ParseOptions *po) {
}

bool ProviderConfig::Validate() const {
if (device < 0) {
SHERPA_ONNX_LOGE("device: '%d' is invalid.", device);
return false;
if(provider == "cuda" || provider == "trt") {
manickavela29 marked this conversation as resolved.
Show resolved Hide resolved
if (device < 0) {
SHERPA_ONNX_LOGE("device: '%d' is invalid.", device);
return false;
}
}
return true;
}
Expand Down
2 changes: 0 additions & 2 deletions sherpa-onnx/python/csrc/sherpa-onnx.h
Original file line number Diff line number Diff line change
Expand Up @@ -5,8 +5,6 @@
#ifndef SHERPA_ONNX_PYTHON_CSRC_SHERPA_ONNX_H_
#define SHERPA_ONNX_PYTHON_CSRC_SHERPA_ONNX_H_

#define PYBIND11_DETAILED_ERROR_MESSAGES

#include "pybind11/functional.h"
#include "pybind11/numpy.h"
#include "pybind11/pybind11.h"
Expand Down
1 change: 0 additions & 1 deletion sherpa-onnx/python/sherpa_onnx/__init__.py
Original file line number Diff line number Diff line change
Expand Up @@ -16,7 +16,6 @@
OfflineTtsVitsModelConfig,
OfflineZipformerAudioTaggingModelConfig,
OnlineStream,
ProviderConfig,
SileroVadModelConfig,
SpeakerEmbeddingExtractor,
SpeakerEmbeddingExtractorConfig,
Expand Down
Loading