pytorch
34 строки · 888.0 Байт
1
2
3
4
5
6from caffe2.python import core
7import caffe2.python.hypothesis_test_util as hu
8from hypothesis import given, settings
9import caffe2.python.serialized_test.serialized_test_util as serial
10import hypothesis.strategies as st
11import numpy as np
12
13import unittest
14
15
16class TestCeil(serial.SerializedTestCase):
17
18@given(X=hu.tensor(),
19engine=st.sampled_from(["", "CUDNN"]),
20**hu.gcs)
21@settings(deadline=10000)
22def test_ceil(self, X, gc, dc, engine):
23op = core.CreateOperator("Ceil", ["X"], ["Y"], engine=engine)
24
25def ceil_ref(X):
26return (np.ceil(X),)
27
28self.assertReferenceChecks(
29device_option=gc,
30op=op,
31inputs=[X],
32reference=ceil_ref)
33
34# Check over multiple devices
35self.assertDeviceChecks(dc, op, [X], [0])
36
37
38if __name__ == "__main__":
39unittest.main()
40