Skip to content

Commit 8b15cdc

Browse files
committed
chore: Update setup.py
Signed-off-by: Dheeraj Peri <[email protected]>
1 parent d890b7d commit 8b15cdc

File tree

1 file changed

+6
-4
lines changed

1 file changed

+6
-4
lines changed

py/setup.py

Lines changed: 6 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -357,8 +357,9 @@ def run(self):
357357
"torch_tensorrt.fx.tracer.acc_tracer",
358358
"torch_tensorrt.fx.tracer.dispatch_tracer",
359359
"torch_tensorrt.dynamo",
360-
"torch_tensorrt.dynamo.passes",
361-
"torch_tensorrt.dynamo.tools",
360+
"torch_tensorrt.dynamo.fx_ts_compat",
361+
"torch_tensorrt.dynamo.fx_ts_compat.passes",
362+
"torch_tensorrt.dynamo.fx_ts_compat.tools",
362363
]
363364
package_dir = {
364365
"torch_tensorrt.fx": "torch_tensorrt/fx",
@@ -368,8 +369,9 @@ def run(self):
368369
"torch_tensorrt.fx.tracer.acc_tracer": "torch_tensorrt/fx/tracer/acc_tracer",
369370
"torch_tensorrt.fx.tracer.dispatch_tracer": "torch_tensorrt/fx/tracer/dispatch_tracer",
370371
"torch_tensorrt.dynamo": "torch_tensorrt/dynamo",
371-
"torch_tensorrt.dynamo.passes": "torch_tensorrt/dynamo/passes",
372-
"torch_tensorrt.dynamo.tools": "torch_tensorrt/dynamo/tools",
372+
"torch_tensorrt.dynamo.fx_ts_compat": "torch_tensorrt/dynamo/fx_ts_compat",
373+
"torch_tensorrt.dynamo.fx_ts_compat.passes": "torch_tensorrt/dynamo/fx_ts_compat/passes",
374+
"torch_tensorrt.dynamo.fx_ts_compat.tools": "torch_tensorrt/dynamo/fx_ts_compat/tools",
373375
}
374376

375377
with open("README.md", "r", encoding="utf-8") as fh:

0 commit comments

Comments
 (0)