🐛 [Bug] Some elementwise like floor_divide.Scalar
using float as default scalar type
#1447
Labels
bug
Something isn't working
Bug Description
Elementwise operations like
floor_divide.Scalar
uses float as default scalar type, this may cause error in such a case:The graph is partitioned into torch/tensorrt-interleaved subgraphs, and
floor_divide.Scalar
accepts one INT-type tensor and one INT-type scalar as input, and the output is fed into the subsequent subgraph as its INT-type input.However, currently the
floor_divide.Scalar
converts the scalar into float by default, and then output a float-type tensor, which does not fullfill the requirement of subsequent graph.To Reproduce
Expected behavior
Environment
conda
,pip
,libtorch
, source):Additional context
The text was updated successfully, but these errors were encountered: