@@ -31,8 +31,8 @@ def test_log_out_type(dtype):
31
31
q = get_queue_or_skip ()
32
32
skip_if_dtype_not_supported (dtype , q )
33
33
34
- X = dpt .asarray (0 , dtype = dtype , sycl_queue = q )
35
- expected_dtype = np .log (np .array (0 , dtype = dtype )).dtype
34
+ X = dpt .asarray (1 , dtype = dtype , sycl_queue = q )
35
+ expected_dtype = np .log (np .array (1 , dtype = dtype )).dtype
36
36
expected_dtype = _map_to_device_dtype (expected_dtype , q .sycl_device )
37
37
assert dpt .log (X ).dtype == expected_dtype
38
38
@@ -44,7 +44,7 @@ def test_log_output_contig(dtype):
44
44
45
45
n_seq = 1027
46
46
47
- X = dpt .linspace (0 , 13 , num = n_seq , dtype = dtype , sycl_queue = q )
47
+ X = dpt .linspace (1 , 13 , num = n_seq , dtype = dtype , sycl_queue = q )
48
48
Xnp = dpt .asnumpy (X )
49
49
50
50
Y = dpt .log (X )
@@ -60,7 +60,7 @@ def test_log_output_strided(dtype):
60
60
61
61
n_seq = 2 * 1027
62
62
63
- X = dpt .linspace (0 , 13 , num = n_seq , dtype = dtype , sycl_queue = q )[::- 2 ]
63
+ X = dpt .linspace (1 , 13 , num = n_seq , dtype = dtype , sycl_queue = q )[::- 2 ]
64
64
Xnp = dpt .asnumpy (X )
65
65
66
66
Y = dpt .log (X )
@@ -119,11 +119,15 @@ def test_log_special_cases():
119
119
q = get_queue_or_skip ()
120
120
121
121
X = dpt .asarray (
122
- [dpt .nan , - 1.0 , 0.0 , - 0.0 , dpt .inf , - dpt .inf ], dtype = "f4" , sycl_queue = q
122
+ [dpt .nan , - dpt .inf , - 1.0 , - 0.0 , 0.0 , dpt .inf ], dtype = "f4" , sycl_queue = q
123
+ )
124
+ Y = dpt .log (X )
125
+
126
+ expected = np .array (
127
+ [np .nan , np .nan , np .nan , - np .inf , - np .inf , np .inf ], dtype = "f4"
123
128
)
124
- Xnp = dpt .asnumpy (X )
125
129
126
- assert_equal (dpt .asnumpy (dpt . log ( X )), np . log ( Xnp ) )
130
+ assert_equal (dpt .asnumpy (Y ), expected )
127
131
128
132
129
133
@pytest .mark .parametrize ("dtype" , ["f2" , "f4" , "f8" , "c8" , "c16" ])
0 commit comments