Skip to content

[CINN] Enable check_cinn on some tests #54261

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
merged 3 commits into from
Jun 5, 2023
Merged

[CINN] Enable check_cinn on some tests #54261

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
3 commits
Select commit Hold shift + click to select a range
55db80b
[CINN] Enable check_cinn
zzk0 Jun 1, 2023
098fadc
Merge remote-tracking branch 'origin/develop' into cinn_enable_test
zzk0 Jun 1, 2023
b6bc35b
add CMakeLists.txt
zzk0 Jun 2, 2023
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
8 changes: 7 additions & 1 deletion test/legacy_test/CMakeLists.txt
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -1187,7 +1187,13 @@ set(TEST_CINN_OPS
test_tile_op
test_roll_op
test_sum_op
test_elementwise_min_op)
test_elementwise_min_op
test_arg_min_max_op
test_reverse_op
test_flip
test_triangular_solve_op
test_scatter_nd_op
test_strided_slice_op)

foreach(TEST_CINN_OPS ${TEST_CINN_OPS})
if(WITH_CINN)
Expand Down
2 changes: 1 addition & 1 deletion test/legacy_test/test_arg_min_max_op.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -42,7 +42,7 @@ def setUp(self):
self.outputs = {'Out': np.argmax(self.x, axis=self.axis)}

def test_check_output(self):
self.check_output()
self.check_output(check_cinn=True)


class TestCase0(BaseTestCase):
Expand Down
8 changes: 4 additions & 4 deletions test/legacy_test/test_flip.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -100,10 +100,10 @@ def init_attrs(self):
self.attrs = {"axis": self.axis}

def test_check_output(self):
self.check_output()
self.check_output(check_cinn=True)

def test_check_grad(self):
self.check_grad(["X"], "Out")
self.check_grad(["X"], "Out", check_cinn=True)

def init_test_case(self):
self.in_shape = (6, 4, 2, 3)
Expand Down Expand Up @@ -167,12 +167,12 @@ def test_check_output(self):
if core.is_compiled_with_cuda():
place = core.CUDAPlace(0)
if core.is_float16_supported(place):
self.check_output_with_place(place)
self.check_output_with_place(place, check_cinn=True)

def test_check_grad(self):
place = core.CUDAPlace(0)
if core.is_float16_supported(place):
self.check_grad_with_place(place, ["X"], "Out")
self.check_grad_with_place(place, ["X"], "Out", check_cinn=True)

cls_name = "{}_{}".format(parent.__name__, "FP16OP")
TestFlipFP16.__name__ = cls_name
Expand Down
4 changes: 2 additions & 2 deletions test/legacy_test/test_reverse_op.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -37,10 +37,10 @@ def setUp(self):
self.outputs = {'Out': out}

def test_check_output(self):
self.check_output()
self.check_output(check_cinn=True)

def test_check_grad(self):
self.check_grad(['X'], 'Out')
self.check_grad(['X'], 'Out', check_cinn=True)


class TestCase0(TestReverseOp):
Expand Down
8 changes: 4 additions & 4 deletions test/legacy_test/test_scatter_nd_op.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -93,7 +93,7 @@ def _set_dtype(self):
self.dtype = np.float64

def test_check_output(self):
self.check_output()
self.check_output(check_cinn=True)

def test_check_grad(self):
self.check_grad(['X', 'Updates'], 'Out', check_prim=True)
Expand Down Expand Up @@ -169,7 +169,7 @@ def _set_dtype(self):
self.dtype = np.float64

def test_check_output(self):
self.check_output()
self.check_output(check_cinn=True)

def test_check_grad(self):
self.check_grad(['X', 'Updates'], 'Out', check_prim=True)
Expand Down Expand Up @@ -248,7 +248,7 @@ def _set_dtype(self):
self.dtype = np.float64

def test_check_output(self):
self.check_output()
self.check_output(check_cinn=True)

def test_check_grad(self):
self.check_grad(['X', 'Updates'], 'Out', check_prim=True)
Expand Down Expand Up @@ -311,7 +311,7 @@ def setUp(self):
self.outputs = {'Out': expect_np}

def test_check_output(self):
self.check_output()
self.check_output(check_cinn=True)

def test_check_grad(self):
self.check_grad(['X', 'Updates'], 'Out', check_prim=True)
Expand Down
8 changes: 4 additions & 4 deletions test/legacy_test/test_strided_slice_op.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -96,10 +96,10 @@ def setUp(self):
}

def test_check_output(self):
self.check_output()
self.check_output(check_cinn=True)

def test_check_grad(self):
self.check_grad({'Input'}, 'Out')
self.check_grad({'Input'}, 'Out', check_cinn=True)

def initTestCase(self):
self.input = np.random.rand(100)
Expand Down Expand Up @@ -1032,10 +1032,10 @@ def setUp(self):
}

def test_check_output(self):
self.check_output()
self.check_output(check_cinn=True)

def test_check_grad(self):
self.check_grad({'Input'}, 'Out')
self.check_grad({'Input'}, 'Out', check_cinn=True)

def initTestCase(self):
self.input = np.random.rand(100)
Expand Down
4 changes: 2 additions & 2 deletions test/legacy_test/test_triangular_solve_op.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -64,10 +64,10 @@ def setUp(self):
self.outputs = {'Out': self.output}

def test_check_output(self):
self.check_output()
self.check_output(check_cinn=True)

def test_check_grad_normal(self):
self.check_grad(['X', 'Y'], 'Out')
self.check_grad(['X', 'Y'], 'Out', check_cinn=True)


# 2D(broadcast) + 3D, test 'transpose'
Expand Down