File tree 2 files changed +7
-5
lines changed
2 files changed +7
-5
lines changed Original file line number Diff line number Diff line change @@ -15,7 +15,7 @@ cc_library(
15
15
srcs = [
16
16
"lowering.cpp" ,
17
17
"drop_unused_nodes.cpp" ,
18
- "register_const_op .cpp"
18
+ "register_trt_placeholder_ops .cpp"
19
19
],
20
20
deps = [
21
21
"//core/lowering/passes" ,
Original file line number Diff line number Diff line change @@ -7,15 +7,17 @@ c10::AliasAnalysisKind aliasAnalysisFromSchema() {
7
7
return c10::AliasAnalysisKind::FROM_SCHEMA;
8
8
}
9
9
10
- // / Op marks a Tensor to be conveted from an Torch Tensor
11
- // / to a TRT constant Tensor
12
- RegisterOperators trt_const_op_reg ({
10
+ RegisterOperators trt_placeholder_ops_reg ({
11
+ // / Op marks a Tensor to be conveted from an Torch Tensor
12
+ // / to a TRT constant Tensor
13
13
Operator (
14
14
" trt::const(Tensor val) -> Tensor" ,
15
15
[](Stack& stack) {
16
16
return 0 ; // noop
17
17
},
18
- aliasAnalysisFromSchema ())});
18
+ aliasAnalysisFromSchema ()),
19
+ });
20
+
19
21
20
22
} // namespace jit
21
23
} // namespace torch
You can’t perform that action at this time.
0 commit comments