Skip to content

Commit 1a4084b

Browse files
committed
chore: update usage in py api
Signed-off-by: Naren Dasan <[email protected]> Signed-off-by: Naren Dasan <[email protected]>
1 parent 85f8cc3 commit 1a4084b

File tree

1 file changed

+4
-4
lines changed

1 file changed

+4
-4
lines changed

py/torch_tensorrt/csrc/tensorrt_classes.cpp

+4-4
Original file line numberDiff line numberDiff line change
@@ -313,10 +313,10 @@ core::CompileSpec CompileSpec::toInternalCompileSpec() {
313313
info.convert_info.engine_settings.device.gpu_id = device.gpu_id;
314314
info.convert_info.engine_settings.device.dla_core = device.dla_core;
315315
info.convert_info.engine_settings.device.allow_gpu_fallback = device.allow_gpu_fallback;
316-
info.partition_info.enabled = torch_fallback.enabled;
317-
info.partition_info.min_block_size = torch_fallback.min_block_size;
318-
info.partition_info.forced_fallback_operators = torch_fallback.forced_fallback_operators;
319-
info.partition_info.truncate_long_and_double = truncate_long_and_double;
316+
info.partitioning_info.enabled = torch_fallback.enabled;
317+
info.partitioning_info.min_block_size = torch_fallback.min_block_size;
318+
info.partitioning_info.forced_fallback_operators = torch_fallback.forced_fallback_operators;
319+
info.partitioning_info.truncate_long_and_double = truncate_long_and_double;
320320
info.lower_info.forced_fallback_modules = torch_fallback.forced_fallback_modules;
321321
info.convert_info.engine_settings.truncate_long_and_double = truncate_long_and_double;
322322

0 commit comments

Comments
 (0)