Skip to content

Commit

Permalink
fix atol
Browse files Browse the repository at this point in the history
  • Loading branch information
cyber-pioneer committed Jan 10, 2024
1 parent 8f29fc6 commit 3f33173
Show file tree
Hide file tree
Showing 3 changed files with 14 additions and 14 deletions.
4 changes: 2 additions & 2 deletions test/legacy_test/CMakeLists.txt
Original file line number Diff line number Diff line change
Expand Up @@ -995,8 +995,8 @@ else()
set_tests_properties(test_conv3d_transpose_part2_op PROPERTIES TIMEOUT 120)
set_tests_properties(test_conv3d_op PROPERTIES TIMEOUT 120)
set_tests_properties(test_norm_op PROPERTIES TIMEOUT 150)
set_tests_properties(test_batch_norm_op_prim_nchw PROPERTIES TIMEOUT 250)
set_tests_properties(test_batch_norm_op_prim_nhwc PROPERTIES TIMEOUT 250)
set_tests_properties(test_batch_norm_op_prim_nchw PROPERTIES TIMEOUT 300)
set_tests_properties(test_batch_norm_op_prim_nhwc PROPERTIES TIMEOUT 300)
set_tests_properties(test_layer_norm_op PROPERTIES TIMEOUT 250)
set_tests_properties(test_pool3d_op PROPERTIES TIMEOUT 150)
endif()
Expand Down
12 changes: 6 additions & 6 deletions test/legacy_test/test_batch_norm_op_prim_nchw.py
Original file line number Diff line number Diff line change
Expand Up @@ -331,12 +331,12 @@ def initConfig(self):
)
class TestBatchNormOpNCHWbf16(TestBatchNormOp):
def initConfig(self):
self.fw_comp_atol = 1e-3
self.fw_comp_rtol = 1e-3
self.rev_comp_atol = 1e-3
self.rev_comp_rtol = 1e-3
self.cinn_atol = 1e-3
self.cinn_rtol = 1e-3
self.fw_comp_atol = 2e-3
self.fw_comp_rtol = 2e-3
self.rev_comp_atol = 2e-3
self.rev_comp_rtol = 2e-3
self.cinn_atol = 2e-3
self.cinn_rtol = 2e-3
self.dtype = "uint16"
self.shape = [16, 16, 16, 8]
self.training = True
Expand Down
12 changes: 6 additions & 6 deletions test/legacy_test/test_batch_norm_op_prim_nhwc.py
Original file line number Diff line number Diff line change
Expand Up @@ -148,12 +148,12 @@ def initConfig(self):
)
class TestBatchNormOpNHWCbf16(TestBatchNormOp):
def initConfig(self):
self.fw_comp_atol = 1e-3
self.fw_comp_rtol = 1e-3
self.rev_comp_atol = 1e-3
self.rev_comp_rtol = 1e-3
self.cinn_atol = 1e-3
self.cinn_rtol = 1e-3
self.fw_comp_atol = 2e-3
self.fw_comp_rtol = 2e-3
self.rev_comp_atol = 2e-3
self.rev_comp_rtol = 2e-3
self.cinn_atol = 2e-3
self.cinn_rtol = 2e-3
self.dtype = "uint16"
self.shape = [16, 16, 16, 8]
self.training = True
Expand Down

0 comments on commit 3f33173

Please sign in to comment.