Skip to content

Commit

Permalink
Fix test missing target (pytorch#73415)
Browse files Browse the repository at this point in the history
Summary:
Pull Request resolved: pytorch#73415

Fix test missing TARGET file issue.

Test Plan: buck test mode/opt deeplearning/trt/fx2trt_oss/test/quant:test_quant_trt

Reviewed By: yinghai

Differential Revision: D34400710

fbshipit-source-id: e68145b4e70db5333f4a8d11a2d240a2f38b4077
(cherry picked from commit fe78e63)
  • Loading branch information
jerryzh168 authored and pytorchmergebot committed Feb 26, 2022
1 parent 0ba3498 commit 186ef8b
Showing 1 changed file with 5 additions and 5 deletions.
10 changes: 5 additions & 5 deletions torch/ao/quantization/fx/backend_config/tensorrt.py
Original file line number Diff line number Diff line change
Expand Up @@ -4,7 +4,7 @@
import torch.nn.intrinsic as nni
import torch.nn.intrinsic.qat as nniqat

from ...fuser_method_mappings import reverse2
from ...fuser_method_mappings import reverse_sequential_wrapper2

def get_tensorrt_backend_config_dict():
""" Get the backend config dictionary for tensorrt backend
Expand Down Expand Up @@ -63,15 +63,15 @@ def get_tensorrt_backend_config_dict():
"dtype_configs": [
weighted_op_qint8_dtype_config,
],
"fuser_method": reverse2(nni.LinearReLU),
"fuser_method": reverse_sequential_wrapper2(nni.LinearReLU),
}
linear_relu_mf_config = {
"pattern": (torch.nn.functional.relu, torch.nn.Linear),
"observation_type": ObservationType.OUTPUT_USE_DIFFERENT_OBSERVER_AS_INPUT,
"dtype_configs": [
weighted_op_qint8_dtype_config,
],
"fuser_method": reverse2(nni.LinearReLU),
"fuser_method": reverse_sequential_wrapper2(nni.LinearReLU),
}

linear_relu_fused_config = {
Expand Down Expand Up @@ -156,15 +156,15 @@ def get_tensorrt_backend_config_dict():
"dtype_configs": [
weighted_op_qint8_dtype_config,
],
"fuser_method": reverse2(nni.ConvReLU2d),
"fuser_method": reverse_sequential_wrapper2(nni.ConvReLU2d),
}
conv2d_relu_mm_config = {
"pattern": (torch.nn.ReLU, torch.nn.Conv2d),
"observation_type": ObservationType.OUTPUT_USE_DIFFERENT_OBSERVER_AS_INPUT,
"dtype_configs": [
weighted_op_qint8_dtype_config,
],
"fuser_method": reverse2(nni.ConvReLU2d),
"fuser_method": reverse_sequential_wrapper2(nni.ConvReLU2d),
}
addmm_config = {
"pattern": torch.addmm,
Expand Down

0 comments on commit 186ef8b

Please sign in to comment.