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 1101a29 commit d3609f8Copy full SHA for d3609f8
py/torch_tensorrt/fx/setup/setup.py
@@ -0,0 +1,22 @@
1
+from setuptools import setup
2
+
3
+setup(
4
+ name="torch_tensorrt_fx2trt",
5
+ version="0.1",
6
+ description="Torch-FX to TensorRT Converter",
7
+ author="PyTorch Team",
8
+ packages=[
9
+ "torch_tensorrt.fx",
10
+ "torch_tensorrt.fx.converters",
11
+ "torch_tensorrt.fx.passes",
12
+ "torch_tensorrt.fx.tools",
13
+ "torch_tensorrt.fx.tracer.acc_tracer",
14
+ ],
15
+ package_dir={
16
+ "torch_tensorrt.fx": "../",
17
+ "torch_tensorrt.fx.converters": "../converters",
18
+ "torch_tensorrt.fx.passes": "../passes",
19
+ "torch_tensorrt.fx.tools": "../tools",
20
+ "torch_tensorrt.fx.tracer.acc_tracer": "../tracer/acc_tracer",
21
+ },
22
+)
0 commit comments