Skip to content

Commit 8b1221b

Browse files
committed
chore: update list of supported ops
Signed-off-by: Naren Dasan <[email protected]> Signed-off-by: Naren Dasan <[email protected]>
1 parent fc6d5ff commit 8b1221b

File tree

5 files changed

+8
-5
lines changed

5 files changed

+8
-5
lines changed

docsrc/indices/supported_ops.rst

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -168,6 +168,7 @@ Operators Currently Supported Through Evaluators
168168
- aten::Int.float(float a) -> int
169169
- aten::Int.int(int a) -> int
170170
- aten::__and__(int a, int b) -> (bool)
171+
- aten::__and__.bool(bool a, bool b) -> (bool)
171172
- aten::__getitem__.t(t[](a) list, int idx) -> (t(*))
172173
- aten::__is__(t1 self, t2 obj) -> bool
173174
- aten::__isnot__(t1 self, t2 obj) -> bool
@@ -195,6 +196,7 @@ Operators Currently Supported Through Evaluators
195196
- aten::eq.float_int(float a, int b) -> (bool)
196197
- aten::eq.int(int a, int b) -> (bool)
197198
- aten::eq.int_float(int a, float b) -> (bool)
199+
- aten::eq.str(str a, str b) -> (bool)
198200
- aten::floor.float(float a) -> (int)
199201
- aten::floor.int(int a) -> (int)
200202
- aten::floordiv.float(float a, float b) -> (int)

tools/linter/utils.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -5,7 +5,8 @@
55

66
BLACKLISTED_BAZEL_TARGETS = [
77
"//experiments", "//tools", "//docker", "//third_party", "//bazel-bin", "//bazel-genfiles",
8-
"//bazel-out", "//bazel-TRTorch", "//bazel-workspace", "//bazel-testlogs", "//py/build",
8+
"//bazel-out", "//bazel-TRTorch", "//bazel-Torch-TensorRT", "//bazel-torch-tensorrt", "//bazel-workspace",
9+
"//bazel-testlogs", "//py/build",
910
"//py/dist", "//py/trtorch.egg-info", "//py/wheelhouse", "//examples", "//docsrc", "//docs"
1011
]
1112

tools/supportedops/BUILD

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -7,6 +7,6 @@ cc_binary(
77
],
88
deps = [
99
"//core/conversion/converters",
10-
"//cpp:trtorch",
10+
"//cpp:torch_tensorrt",
1111
],
1212
)

tools/supportedops/main.cpp

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -7,8 +7,8 @@
77
#include <vector>
88

99
int main(int argc, const char* argv[]) {
10-
std::vector<std::string> converters = trtorch::core::conversion::converters::get_converter_list();
11-
std::vector<std::string> evaluators = trtorch::core::conversion::evaluators::getEvaluatorList();
10+
std::vector<std::string> converters = torch_tensorrt::core::conversion::converters::get_converter_list();
11+
std::vector<std::string> evaluators = torch_tensorrt::core::conversion::evaluators::getEvaluatorList();
1212

1313
std::stringstream ss;
1414

tools/trtorchexec/BUILD

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -6,7 +6,7 @@ cc_binary(
66
"main.cpp",
77
],
88
deps = [
9-
"//cpp/api:trtorch",
9+
"//cpp:torch_tensorrt",
1010
"@libtorch",
1111
"@libtorch//:caffe2",
1212
],

0 commit comments

Comments
 (0)