def test_hermitian_float64(self): # M < N a = np.array([[1, 2, 3, 4, 5, 6], [7, 8, 9, 10, 11, 12]], np.float64) a_gpu = gpuarray.to_gpu(a) at_gpu = linalg.hermitian(a_gpu) assert np.all(a.T == at_gpu.get()) # M > N b = a.T.copy() b_gpu = gpuarray.to_gpu(b) bt_gpu = linalg.hermitian(b_gpu) assert np.all(b.T == bt_gpu.get())
def test_hermitian_complex128(self): # M < N a = np.array([[1j, 2j, 3j, 4j, 5j, 6j], [7j, 8j, 9j, 10j, 11j, 12j]], np.complex128) a_gpu = gpuarray.to_gpu(a) at_gpu = linalg.hermitian(a_gpu) assert np.all(np.conj(a.T) == at_gpu.get()) # M > N b = a.T.copy() b_gpu = gpuarray.to_gpu(b) bt_gpu = linalg.hermitian(b_gpu) assert np.all(np.conj(b.T) == bt_gpu.get())