-
Notifications
You must be signed in to change notification settings - Fork 369
Closed
Labels
bugSomething isn't workingSomething isn't workingcomponent: convertersIssues re: Specific op convertersIssues re: Specific op converters
Description
Bug Description
The current implementation of the aten::abs converter relies on the UnaryLayer kABS implementation which does not support integers
auto unary = ctx->net->addUnary(*in, nvinfer1::UnaryOperation::trt_type); \ |
Generally the input must have a floating-point type (or kINT8 as a quantized float), except for the following operations:
kSIGN accepts a floating-point or Int32 tensor.
kNOT requires a Bool tensor.
To Reproduce
Steps to reproduce the behavior:
- Attempt to compile a model with an aten::abs op with integer inputs.
Expected behavior
This can be supported with an element-wise implementation of the op in cases where the unarylayer does not support the input type. abs(x) = max(x, x*-1)
Environment
Build information about Torch-TensorRT can be found by turning on debug messages
- Torch-TensorRT Version (e.g. 1.0.0):
- PyTorch Version (e.g. 1.0):
- CPU Architecture:
- OS (e.g., Linux):
- How you installed PyTorch (
conda
,pip
,libtorch
, source): - Build command you used (if compiling from source):
- Are you using local sources or building from archives:
- Python version:
- CUDA version:
- GPU models and configuration:
- Any other relevant information:
Additional context
Metadata
Metadata
Labels
bugSomething isn't workingSomething isn't workingcomponent: convertersIssues re: Specific op convertersIssues re: Specific op converters