File tree Expand file tree Collapse file tree 1 file changed +3
-0
lines changed Expand file tree Collapse file tree 1 file changed +3
-0
lines changed Original file line number Diff line number Diff line change @@ -73,6 +73,7 @@ def test_compile_script(self):
73
73
same = (trt_mod (self .input ) - self .scripted_model (self .input )).abs ().max ()
74
74
self .assertTrue (same < 2e-2 )
75
75
76
+
76
77
class TestCompileHalf (ModelTestCase ):
77
78
78
79
def setUp (self ):
@@ -94,6 +95,7 @@ def test_compile_script_half(self):
94
95
same = (trt_mod (self .input .half ()) - self .scripted_model (self .input .half ())).abs ().max ()
95
96
self .assertTrue (same < 2e-2 )
96
97
98
+
97
99
class TestCompileHalfDefault (ModelTestCase ):
98
100
99
101
def setUp (self ):
@@ -115,6 +117,7 @@ def test_compile_script_half_by_default(self):
115
117
same = (trt_mod (self .input .half ()) - self .scripted_model (self .input .half ())).abs ().max ()
116
118
self .assertTrue (same < 2e-2 )
117
119
120
+
118
121
class TestFallbackToTorch (ModelTestCase ):
119
122
120
123
def setUp (self ):
You can’t perform that action at this time.
0 commit comments