Toggle navigation
Toggle navigation
此项目
正在载入...
Sign in
xuning
/
sherpaonnx
转到一个项目
Toggle navigation
项目
群组
代码片段
帮助
Toggle navigation pinning
Project
Activity
Repository
Pipelines
Graphs
Issues
0
Merge Requests
0
Wiki
Network
Create a new issue
Builds
Commits
Authored by
Manix
2024-07-08 18:08:16 +0530
Browse Files
Options
Browse Files
Download
Email Patches
Plain Diff
Committed by
GitHub
2024-07-08 20:38:16 +0800
Commit
3e4307e2fb88d4b1b648211c14f2fff6db11bca4
3e4307e2
1 parent
4fd04930
updating trt workspace int64 (#1094)
Signed-off-by: Manix <manickavela1998@gmail.com>
显示空白字符变更
内嵌
并排对比
正在显示
3 个修改的文件
包含
4 行增加
和
4 行删除
sherpa-onnx/csrc/provider-config.cc
sherpa-onnx/csrc/provider-config.h
sherpa-onnx/python/csrc/tensorrt-config.cc
sherpa-onnx/csrc/provider-config.cc
查看文件 @
3e4307e
...
...
@@ -60,7 +60,7 @@ void TensorrtConfig::Register(ParseOptions *po) {
bool
TensorrtConfig
::
Validate
()
const
{
if
(
trt_max_workspace_size
<
0
)
{
SHERPA_ONNX_LOGE
(
"trt_max_workspace_size: %d is not valid."
,
SHERPA_ONNX_LOGE
(
"trt_max_workspace_size: %
ll
d is not valid."
,
trt_max_workspace_size
);
return
false
;
}
...
...
sherpa-onnx/csrc/provider-config.h
查看文件 @
3e4307e
...
...
@@ -27,7 +27,7 @@ struct CudaConfig {
};
struct
TensorrtConfig
{
int
32
_t
trt_max_workspace_size
=
2147483647
;
int
64
_t
trt_max_workspace_size
=
2147483647
;
int32_t
trt_max_partition_iterations
=
10
;
int32_t
trt_min_subgraph_size
=
5
;
bool
trt_fp16_enable
=
true
;
...
...
@@ -39,7 +39,7 @@ struct TensorrtConfig {
bool
trt_dump_subgraphs
=
false
;
TensorrtConfig
()
=
default
;
TensorrtConfig
(
int
32
_t
trt_max_workspace_size
,
TensorrtConfig
(
int
64
_t
trt_max_workspace_size
,
int32_t
trt_max_partition_iterations
,
int32_t
trt_min_subgraph_size
,
bool
trt_fp16_enable
,
...
...
sherpa-onnx/python/csrc/tensorrt-config.cc
查看文件 @
3e4307e
...
...
@@ -14,7 +14,7 @@ void PybindTensorrtConfig(py::module *m) {
using
PyClass
=
TensorrtConfig
;
py
::
class_
<
PyClass
>
(
*
m
,
"TensorrtConfig"
)
.
def
(
py
::
init
<>
())
.
def
(
py
::
init
([](
int
32
_t
trt_max_workspace_size
,
.
def
(
py
::
init
([](
int
64
_t
trt_max_workspace_size
,
int32_t
trt_max_partition_iterations
,
int32_t
trt_min_subgraph_size
,
bool
trt_fp16_enable
,
...
...
请
注册
或
登录
后发表评论