File tree Expand file tree Collapse file tree 1 file changed +1
-3
lines changed
core/conversion/evaluators Expand file tree Collapse file tree 1 file changed +1
-3
lines changed Original file line number Diff line number Diff line change @@ -470,8 +470,6 @@ auto aten_registrations TRTORCH_UNUSED =
470
470
EvalOptions ()})
471
471
.evaluator({c10::Symbol::fromQualString (" aten::arange" ),
472
472
[](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());
475
473
int input_size = n->inputs ().size ();
476
474
int scalar_count = 0 ;
477
475
for (int i = 0 ; i < input_size; i++) {
@@ -484,7 +482,7 @@ auto aten_registrations TRTORCH_UNUSED =
484
482
int end_scalar = args.at (n->input (0 )).unwrapToInt ();
485
483
return torch::arange (end_scalar);
486
484
} 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 >();
488
486
return torch::arange (end_scalar);
489
487
}
490
488
} else if (scalar_count == 2 ) {
You can’t perform that action at this time.
0 commit comments