Remove TRT 4.0 restrictions on int32 test.
PiperOrigin-RevId: 249264857
This commit is contained in:
parent
9d46f5599f
commit
bcb89187a3
@ -71,12 +71,6 @@ class ExcludeUnsupportedInt32Test(trt_test.TfTrtIntegrationTestBase):
|
||||
"""Return the expected engines to build."""
|
||||
return []
|
||||
|
||||
def ShouldRunTest(self, run_params):
|
||||
"""Whether to run the test."""
|
||||
# TODO(aaroey): Trt 4.0 forbids conversion for tensors with rank <3 in int8
|
||||
# mode, which is a bug. Re-enable this when trt library is fixed.
|
||||
return not trt_test.IsQuantizationMode(run_params.precision_mode)
|
||||
|
||||
|
||||
if __name__ == '__main__':
|
||||
test.main()
|
||||
|
Loading…
Reference in New Issue
Block a user