Skip to content

Commit f7b6c9a

Browse files
committed
merge main to the branch
1 parent 91d9c38 commit f7b6c9a

File tree

4 files changed

+3
-1
lines changed

4 files changed

+3
-1
lines changed

py/torch_tensorrt/csrc/tensorrt_classes.cpp

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -349,7 +349,6 @@ core::CompileSpec CompileSpec::toInternalCompileSpec(bool converting_to_trt_engi
349349
info.lower_info.unfreeze_module = true;
350350
info.lower_info.disable_cse = true;
351351
}
352-
353352
info.convert_info.engine_settings.sparse_weights = sparse_weights;
354353
info.convert_info.engine_settings.disable_tf32 = disable_tf32;
355354
info.convert_info.engine_settings.refit = refit;

py/torch_tensorrt/fx/test/converters/acc_op/test_dequantize.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -3,6 +3,7 @@
33
import tensorrt as trt
44
import torch.fx
55
import torch.nn as nn
6+
67
import torch_tensorrt.fx.tracer.acc_tracer.acc_ops as acc_ops
78
from torch.testing._internal.common_utils import run_tests
89
from torch_tensorrt.fx.tools.common_fx2trt import AccTestCase, InputTensorSpec

py/torch_tensorrt/fx/test/converters/acc_op/test_pad.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -3,6 +3,7 @@
33
import tensorrt as trt
44
import torch
55
import torch.nn as nn
6+
67
import torch_tensorrt.fx.tracer.acc_tracer.acc_ops as acc_ops
78
from parameterized import parameterized
89
from torch.testing._internal.common_utils import run_tests

py/torch_tensorrt/fx/test/converters/acc_op/test_quantize_per_tensor.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -3,6 +3,7 @@
33
import tensorrt as trt
44
import torch.fx
55
import torch.nn as nn
6+
67
import torch_tensorrt.fx.tracer.acc_tracer.acc_ops as acc_ops
78
from torch.testing._internal.common_utils import run_tests
89
from torch_tensorrt.fx.tools.common_fx2trt import AccTestCase, InputTensorSpec

0 commit comments

Comments
 (0)