@@ -357,8 +357,9 @@ def run(self):
357
357
"torch_tensorrt.fx.tracer.acc_tracer" ,
358
358
"torch_tensorrt.fx.tracer.dispatch_tracer" ,
359
359
"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" ,
362
363
]
363
364
package_dir = {
364
365
"torch_tensorrt.fx" : "torch_tensorrt/fx" ,
@@ -368,8 +369,9 @@ def run(self):
368
369
"torch_tensorrt.fx.tracer.acc_tracer" : "torch_tensorrt/fx/tracer/acc_tracer" ,
369
370
"torch_tensorrt.fx.tracer.dispatch_tracer" : "torch_tensorrt/fx/tracer/dispatch_tracer" ,
370
371
"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" ,
373
375
}
374
376
375
377
with open ("README.md" , "r" , encoding = "utf-8" ) as fh :
0 commit comments