Skip to content

Commit d19682b

Browse files
committed
refactor: remove ceil and comment for aten::arange
Signed-off-by: inocsin <[email protected]>
1 parent 014e381 commit d19682b

File tree

1 file changed

+1
-3
lines changed

1 file changed

+1
-3
lines changed

core/conversion/evaluators/aten.cpp

Lines changed: 1 addition & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -470,8 +470,6 @@ auto aten_registrations TRTORCH_UNUSED =
470470
EvalOptions()})
471471
.evaluator({c10::Symbol::fromQualString("aten::arange"),
472472
[](const torch::jit::Node* n, kwargs& args) -> c10::optional<torch::jit::IValue> {
473-
// int end_scalar = 0;
474-
// auto end_scalar = ceil(args.at(n->input(0)).unwrapToScalar());
475473
int input_size = n->inputs().size();
476474
int scalar_count = 0;
477475
for (int i = 0; i < input_size; i++) {
@@ -484,7 +482,7 @@ auto aten_registrations TRTORCH_UNUSED =
484482
int end_scalar = args.at(n->input(0)).unwrapToInt();
485483
return torch::arange(end_scalar);
486484
} else if (args.at(n->input(0)).IValue()->isDouble()) {
487-
float end_scalar = ceil(args.at(n->input(0)).unwrapToScalar().to<float>());
485+
float end_scalar = args.at(n->input(0)).unwrapToScalar().to<float>();
488486
return torch::arange(end_scalar);
489487
}
490488
} else if (scalar_count == 2) {

0 commit comments

Comments
 (0)