File tree Expand file tree Collapse file tree 2 files changed +1
-2
lines changed
core/conversion/evaluators Expand file tree Collapse file tree 2 files changed +1
-2
lines changed Original file line number Diff line number Diff line change @@ -238,7 +238,7 @@ at::Tensor createTensorFromList(
238
238
// / Gets shape of tensor to be created
239
239
auto sizes = compute_sizes (data);
240
240
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);
242
242
if (initial_scalar_type == at::ScalarType::Double) {
243
243
initial_scalar_type = at::typeMetaToScalarType (c10::get_default_dtype ());
244
244
}
Original file line number Diff line number Diff line change @@ -39,7 +39,6 @@ FROM base as torch-tensorrt
39
39
# copy source repo
40
40
COPY . /workspace/torch_tensorrt
41
41
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
43
42
RUN conda init bash
44
43
45
44
RUN pip3 install ipywidgets --trusted-host pypi.org --trusted-host pypi.python.org --trusted-host=files.pythonhosted.org
You can’t perform that action at this time.
0 commit comments