We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 8d8cbfd commit 13cc024Copy full SHA for 13cc024
py/torch_tensorrt/ts/_compile_spec.py
@@ -226,7 +226,13 @@ def _parse_input_signature(input_signature: Any):
226
227
def _parse_compile_spec(compile_spec_: Dict[str, Any]) -> _ts_C.CompileSpec:
228
# TODO: Use deepcopy to support partial compilation of collections
229
- compile_spec = compile_spec_
+ compile_spec = {}
230
+ for k, v in compile_spec_.items():
231
+ if k != "calibrator":
232
+ compile_spec[k] = deepcopy(v)
233
+ else:
234
+ compile_spec[k] = v
235
+
236
info = _ts_C.CompileSpec()
237
238
if len(compile_spec["inputs"]) > 0:
0 commit comments