gmarkall/example_using_cuda_overload.py

Created October 21, 2022 11:36

Star () You must be signed in to star a gist
Fork () You must be signed in to fork a gist

Learn more about clone URLs
Clone this repository at <script src="https://gist.github.com/gmarkall/d91971ff6fe427d00b7f532086af46bb.js"></script>
Save gmarkall/d91971ff6fe427d00b7f532086af46bb to your computer and use it in GitHub Desktop.

Download ZIP

Example using an overload in the CUDA target

Raw

example_using_cuda_overload.py

	from numba import cuda
	import numpy as np

	@cuda.jit
	def f(arr):
	print("Sum is", arr.sum())

	# Prints "Sum is 10"
	f[1, 1](np.arange(5))

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment