feat: Adding support for native int64 by narendasan · Pull Request #2789 · pytorch/TensorRT (original) (raw)

diff --git a/home/runner/work/TensorRT/TensorRT/core/util/trt_util.cpp b/tmp/changes.txt index 503b88e..3ca5780 100644 --- a/home/runner/work/TensorRT/TensorRT/core/util/trt_util.cpp +++ b/tmp/changes.txt @@ -164,8 +164,8 @@ nvinfer1::Dims unsqueezeDims(const nvinfer1::Dims& d, int pos, int val, bool use // Acceptable range for pos is [-d.nbDims - 1, d.nbDims] TORCHTRT_ASSERT( pos >= (-d.nbDims - 1) && pos <= d.nbDims,

•  "ERROR: Index to unsqueeze is out of bounds. " << "Expected value in range [" << (-d.nbDims - 1) << ", "

•                                                 << d.nbDims << "], but got " << pos);

•  "ERROR: Index to unsqueeze is out of bounds. "

•      << "Expected value in range [" << (-d.nbDims - 1) << ", " << d.nbDims << "], but got " << pos);

  // Unsqueeze with negative dimensions creates a new dimension at that index pos = (pos < 0) ? (pos + d.nbDims + 1) : pos;

ERROR: Some files do not conform to style guidelines