arrayfire-python/tests/simple/interop.py at master · arrayfire/arrayfire-python · GitHub

History

140 lines (115 loc) · 3.69 KB

Raw

100

101

102

103

104

105

106

107

108

109

110

111

112

113

114

115

116

117

118

119

120

121

122

123

124

125

126

127

128

129

130

131

132

133

134

135

136

137

138

139

140

#!/usr/bin/env python

#######################################################

# This file is distributed under 3-clause BSD license.

# The complete license agreement can be obtained at:

# http://arrayfire.com/licenses/BSD-3-Clause

########################################################

import arrayfire as af

from . import _util

def simple_interop(verbose=False):

if af.AF_NUMPY_FOUND:

import numpy as np

n = np.random.random((5,))

a = af.to_array(n)

n2 = np.array(a)

assert((n == n2).all())

n2[:] = 0

a.to_ndarray(n2)

assert((n == n2).all())

n = np.random.random((5, 3))

a = af.to_array(n)

n2 = np.array(a)

assert((n == n2).all())

n2[:] = 0

a.to_ndarray(n2)

assert((n == n2).all())

n = np.random.random((5, 3, 2))

a = af.to_array(n)

n2 = np.array(a)

assert((n == n2).all())

n2[:] = 0

a.to_ndarray(n2)

assert((n == n2).all())

n = np.random.random((5, 3, 2, 2))

a = af.to_array(n)

n2 = np.array(a)

assert((n == n2).all())

n2[:] = 0

a.to_ndarray(n2)

assert((n == n2).all())

if af.AF_PYCUDA_FOUND and af.get_active_backend() == "cuda":

import pycuda.gpuarray as cudaArray

n = np.random.random((5,))

c = cudaArray.to_gpu(n)

a = af.to_array(c)

n2 = np.array(a)

assert((n == n2).all())

n = np.random.random((5, 3))

c = cudaArray.to_gpu(n)

a = af.to_array(c)

n2 = np.array(a)

assert((n == n2).all())

n = np.random.random((5, 3, 2))

c = cudaArray.to_gpu(n)

a = af.to_array(c)

n2 = np.array(a)

assert((n == n2).all())

n = np.random.random((5, 3, 2, 2))

c = cudaArray.to_gpu(n)

a = af.to_array(c)

n2 = np.array(a)

assert((n == n2).all())

if af.AF_PYOPENCL_FOUND and af.backend.name() == "opencl":

# TODO: This needs fixing upstream

# https://github.com/arrayfire/arrayfire/issues/1728

# import pyopencl as cl

# import pyopencl.array as clArray

# ctx = cl.create_some_context()

# queue = cl.CommandQueue(ctx)

# n = np.random.random((5,))

# c = cl.array.to_device(queue, n)

# a = af.to_array(c)

# n2 = np.array(a)

# assert((n==n2).all())

# n = np.random.random((5,3))

# c = cl.array.to_device(queue, n)

# a = af.to_array(c)

# n2 = np.array(a)

# assert((n==n2).all())

# n = np.random.random((5,3,2))

# c = cl.array.to_device(queue, n)

# a = af.to_array(c)

# n2 = np.array(a)

# assert((n==n2).all())

# n = np.random.random((5,3,2,2))

# c = cl.array.to_device(queue, n)

# a = af.to_array(c)

# n2 = np.array(a)

# assert((n==n2).all())

pass

if af.AF_NUMBA_FOUND and af.get_active_backend() == "cuda":

from numba import cuda

n = np.random.random((5,))

c = cuda.to_device(n)

a = af.to_array(c)

n2 = np.array(a)

assert((n == n2).all())

n = np.random.random((5, 3))

c = cuda.to_device(n)

a = af.to_array(c)

n2 = np.array(a)

assert((n == n2).all())

n = np.random.random((5, 3, 2))

c = cuda.to_device(n)

a = af.to_array(c)

n2 = np.array(a)

assert((n == n2).all())

n = np.random.random((5, 3, 2, 2))

c = cuda.to_device(n)

a = af.to_array(c)

n2 = np.array(a)

assert((n == n2).all())

_util.tests["interop"] = simple_interop

arrayfire-python/tests/simple/interop.py at master · arrayfire/arrayfire-python · GitHub

Uh oh!

Expand file tree

Breadcrumbs

interop.py

Latest commit

History

Breadcrumbs

interop.py

File metadata and controls