Support compile on windows x86 (#49)

* Support compile on windows x86

* fix cmake

Co-authored-by: root <root@bjyz-sys-gpu-kongming3.bjyz.baidu.com>
This commit is contained in:
Jason
2022-07-28 11:43:02 +08:00
committed by GitHub
parent adddd3c452
commit a9b5c6ed85
7 changed files with 36 additions and 0 deletions

View File

@@ -292,6 +292,7 @@ TensorInfo OrtBackend::GetOutputInfo(int index) {
}
void OrtBackend::InitCustomOperators() {
#ifndef NON_64_PLATFORM
if (custom_operators_.size() == 0) {
MultiClassNmsOp* custom_op = new MultiClassNmsOp{};
custom_operators_.push_back(custom_op);
@@ -300,6 +301,7 @@ void OrtBackend::InitCustomOperators() {
custom_op_domain_.Add(custom_operators_[i]);
}
session_options_.Add(custom_op_domain_);
#endif
}
} // namespace fastdeploy