Skip to content

Commit bdd9c58

Browse files
author
Anurag Dixit
committed
(//core): Fixed the evaluator with updated torch from DLFW
Signed-off-by: Anurag Dixit <[email protected]>
1 parent d694dc5 commit bdd9c58

File tree

2 files changed

+1
-2
lines changed

2 files changed

+1
-2
lines changed

core/conversion/evaluators/eval_util.cpp

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -238,7 +238,7 @@ at::Tensor createTensorFromList(
238238
/// Gets shape of tensor to be created
239239
auto sizes = compute_sizes(data);
240240
checkListInputType(elem_type, sizes.size() == 1 && sizes[0] == 0);
241-
at::ScalarType initial_scalar_type = c10::scalarTypeFromJitType(elem_type);
241+
at::ScalarType initial_scalar_type = c10::scalarTypeFromJitType(*elem_type);
242242
if (initial_scalar_type == at::ScalarType::Double) {
243243
initial_scalar_type = at::typeMetaToScalarType(c10::get_default_dtype());
244244
}

docker/Dockerfile

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -39,7 +39,6 @@ FROM base as torch-tensorrt
3939
# copy source repo
4040
COPY . /workspace/torch_tensorrt
4141
COPY --from=torch-tensorrt-builder /workspace/torch_tensorrt/src/py/dist/ .
42-
RUN patch -u /opt/conda/lib/python3.8/site-packages/pytorch_quantization/nn/modules/tensor_quantizer.py -i /workspace/torch_tensorrt/docker/qat.patch
4342
RUN conda init bash
4443

4544
RUN pip3 install ipywidgets --trusted-host pypi.org --trusted-host pypi.python.org --trusted-host=files.pythonhosted.org

0 commit comments

Comments
 (0)