Skip to content

Commit d3609f8

Browse files
author
Wei Wei
committed
add setup
1 parent 1101a29 commit d3609f8

File tree

1 file changed

+22
-0
lines changed

1 file changed

+22
-0
lines changed
Lines changed: 22 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -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

Comments
 (0)