numba-cuda 0.0.1__tar.gz → 0.0.15__tar.gz

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (242) hide show
  1. numba_cuda-0.0.15/LICENSE +25 -0
  2. numba_cuda-0.0.15/PKG-INFO +69 -0
  3. numba_cuda-0.0.15/README.md +54 -0
  4. numba_cuda-0.0.15/numba_cuda/VERSION +1 -0
  5. numba_cuda-0.0.15/numba_cuda/__init__.py +5 -0
  6. numba_cuda-0.0.15/numba_cuda/_version.py +19 -0
  7. numba_cuda-0.0.15/numba_cuda/numba/cuda/__init__.py +22 -0
  8. numba_cuda-0.0.15/numba_cuda/numba/cuda/api.py +526 -0
  9. numba_cuda-0.0.15/numba_cuda/numba/cuda/api_util.py +30 -0
  10. numba_cuda-0.0.15/numba_cuda/numba/cuda/args.py +77 -0
  11. numba_cuda-0.0.15/numba_cuda/numba/cuda/cg.py +62 -0
  12. numba_cuda-0.0.15/numba_cuda/numba/cuda/codegen.py +378 -0
  13. numba_cuda-0.0.15/numba_cuda/numba/cuda/compiler.py +422 -0
  14. numba_cuda-0.0.15/numba_cuda/numba/cuda/cpp_function_wrappers.cu +47 -0
  15. numba_cuda-0.0.15/numba_cuda/numba/cuda/cuda_fp16.h +3631 -0
  16. numba_cuda-0.0.15/numba_cuda/numba/cuda/cuda_fp16.hpp +2465 -0
  17. numba_cuda-0.0.15/numba_cuda/numba/cuda/cuda_paths.py +258 -0
  18. numba_cuda-0.0.15/numba_cuda/numba/cuda/cudadecl.py +806 -0
  19. numba_cuda-0.0.15/numba_cuda/numba/cuda/cudadrv/__init__.py +9 -0
  20. numba_cuda-0.0.15/numba_cuda/numba/cuda/cudadrv/devicearray.py +904 -0
  21. numba_cuda-0.0.15/numba_cuda/numba/cuda/cudadrv/devices.py +248 -0
  22. numba_cuda-0.0.15/numba_cuda/numba/cuda/cudadrv/driver.py +3201 -0
  23. numba_cuda-0.0.15/numba_cuda/numba/cuda/cudadrv/drvapi.py +398 -0
  24. numba_cuda-0.0.15/numba_cuda/numba/cuda/cudadrv/dummyarray.py +452 -0
  25. numba_cuda-0.0.15/numba_cuda/numba/cuda/cudadrv/enums.py +607 -0
  26. numba_cuda-0.0.15/numba_cuda/numba/cuda/cudadrv/error.py +36 -0
  27. numba_cuda-0.0.15/numba_cuda/numba/cuda/cudadrv/libs.py +176 -0
  28. numba_cuda-0.0.15/numba_cuda/numba/cuda/cudadrv/ndarray.py +20 -0
  29. numba_cuda-0.0.15/numba_cuda/numba/cuda/cudadrv/nvrtc.py +260 -0
  30. numba_cuda-0.0.15/numba_cuda/numba/cuda/cudadrv/nvvm.py +707 -0
  31. numba_cuda-0.0.15/numba_cuda/numba/cuda/cudadrv/rtapi.py +10 -0
  32. numba_cuda-0.0.15/numba_cuda/numba/cuda/cudadrv/runtime.py +142 -0
  33. numba_cuda-0.0.15/numba_cuda/numba/cuda/cudaimpl.py +1055 -0
  34. numba_cuda-0.0.15/numba_cuda/numba/cuda/cudamath.py +140 -0
  35. numba_cuda-0.0.15/numba_cuda/numba/cuda/decorators.py +189 -0
  36. numba_cuda-0.0.15/numba_cuda/numba/cuda/descriptor.py +33 -0
  37. numba_cuda-0.0.15/numba_cuda/numba/cuda/device_init.py +89 -0
  38. numba_cuda-0.0.15/numba_cuda/numba/cuda/deviceufunc.py +908 -0
  39. numba_cuda-0.0.15/numba_cuda/numba/cuda/dispatcher.py +1057 -0
  40. numba_cuda-0.0.15/numba_cuda/numba/cuda/errors.py +59 -0
  41. numba_cuda-0.0.15/numba_cuda/numba/cuda/extending.py +7 -0
  42. numba_cuda-0.0.15/numba_cuda/numba/cuda/initialize.py +13 -0
  43. numba_cuda-0.0.15/numba_cuda/numba/cuda/intrinsic_wrapper.py +77 -0
  44. numba_cuda-0.0.15/numba_cuda/numba/cuda/intrinsics.py +198 -0
  45. numba_cuda-0.0.15/numba_cuda/numba/cuda/kernels/reduction.py +262 -0
  46. numba_cuda-0.0.15/numba_cuda/numba/cuda/kernels/transpose.py +65 -0
  47. numba_cuda-0.0.15/numba_cuda/numba/cuda/libdevice.py +3382 -0
  48. numba_cuda-0.0.15/numba_cuda/numba/cuda/libdevicedecl.py +17 -0
  49. numba_cuda-0.0.15/numba_cuda/numba/cuda/libdevicefuncs.py +1057 -0
  50. numba_cuda-0.0.15/numba_cuda/numba/cuda/libdeviceimpl.py +83 -0
  51. numba_cuda-0.0.15/numba_cuda/numba/cuda/mathimpl.py +448 -0
  52. numba_cuda-0.0.15/numba_cuda/numba/cuda/models.py +48 -0
  53. numba_cuda-0.0.15/numba_cuda/numba/cuda/nvvmutils.py +235 -0
  54. numba_cuda-0.0.15/numba_cuda/numba/cuda/printimpl.py +86 -0
  55. numba_cuda-0.0.15/numba_cuda/numba/cuda/random.py +292 -0
  56. numba_cuda-0.0.15/numba_cuda/numba/cuda/simulator/__init__.py +38 -0
  57. numba_cuda-0.0.15/numba_cuda/numba/cuda/simulator/api.py +110 -0
  58. numba_cuda-0.0.15/numba_cuda/numba/cuda/simulator/compiler.py +9 -0
  59. numba_cuda-0.0.15/numba_cuda/numba/cuda/simulator/cudadrv/__init__.py +2 -0
  60. numba_cuda-0.0.15/numba_cuda/numba/cuda/simulator/cudadrv/devicearray.py +432 -0
  61. numba_cuda-0.0.15/numba_cuda/numba/cuda/simulator/cudadrv/devices.py +117 -0
  62. numba_cuda-0.0.15/numba_cuda/numba/cuda/simulator/cudadrv/driver.py +62 -0
  63. numba_cuda-0.0.15/numba_cuda/numba/cuda/simulator/cudadrv/drvapi.py +4 -0
  64. numba_cuda-0.0.15/numba_cuda/numba/cuda/simulator/cudadrv/dummyarray.py +4 -0
  65. numba_cuda-0.0.15/numba_cuda/numba/cuda/simulator/cudadrv/error.py +6 -0
  66. numba_cuda-0.0.15/numba_cuda/numba/cuda/simulator/cudadrv/libs.py +2 -0
  67. numba_cuda-0.0.15/numba_cuda/numba/cuda/simulator/cudadrv/nvvm.py +29 -0
  68. numba_cuda-0.0.15/numba_cuda/numba/cuda/simulator/cudadrv/runtime.py +19 -0
  69. numba_cuda-0.0.15/numba_cuda/numba/cuda/simulator/kernel.py +308 -0
  70. numba_cuda-0.0.15/numba_cuda/numba/cuda/simulator/kernelapi.py +495 -0
  71. numba_cuda-0.0.15/numba_cuda/numba/cuda/simulator/reduction.py +15 -0
  72. numba_cuda-0.0.15/numba_cuda/numba/cuda/simulator/vector_types.py +58 -0
  73. numba_cuda-0.0.15/numba_cuda/numba/cuda/simulator_init.py +17 -0
  74. numba_cuda-0.0.15/numba_cuda/numba/cuda/stubs.py +902 -0
  75. numba_cuda-0.0.15/numba_cuda/numba/cuda/target.py +440 -0
  76. numba_cuda-0.0.15/numba_cuda/numba/cuda/testing.py +202 -0
  77. numba_cuda-0.0.15/numba_cuda/numba/cuda/tests/__init__.py +58 -0
  78. numba_cuda-0.0.15/numba_cuda/numba/cuda/tests/cudadrv/__init__.py +8 -0
  79. numba_cuda-0.0.15/numba_cuda/numba/cuda/tests/cudadrv/test_array_attr.py +145 -0
  80. numba_cuda-0.0.15/numba_cuda/numba/cuda/tests/cudadrv/test_context_stack.py +145 -0
  81. numba_cuda-0.0.15/numba_cuda/numba/cuda/tests/cudadrv/test_cuda_array_slicing.py +375 -0
  82. numba_cuda-0.0.15/numba_cuda/numba/cuda/tests/cudadrv/test_cuda_auto_context.py +21 -0
  83. numba_cuda-0.0.15/numba_cuda/numba/cuda/tests/cudadrv/test_cuda_devicerecord.py +179 -0
  84. numba_cuda-0.0.15/numba_cuda/numba/cuda/tests/cudadrv/test_cuda_driver.py +235 -0
  85. numba_cuda-0.0.15/numba_cuda/numba/cuda/tests/cudadrv/test_cuda_libraries.py +22 -0
  86. numba_cuda-0.0.15/numba_cuda/numba/cuda/tests/cudadrv/test_cuda_memory.py +193 -0
  87. numba_cuda-0.0.15/numba_cuda/numba/cuda/tests/cudadrv/test_cuda_ndarray.py +547 -0
  88. numba_cuda-0.0.15/numba_cuda/numba/cuda/tests/cudadrv/test_deallocations.py +249 -0
  89. numba_cuda-0.0.15/numba_cuda/numba/cuda/tests/cudadrv/test_detect.py +81 -0
  90. numba_cuda-0.0.15/numba_cuda/numba/cuda/tests/cudadrv/test_emm_plugins.py +192 -0
  91. numba_cuda-0.0.15/numba_cuda/numba/cuda/tests/cudadrv/test_events.py +38 -0
  92. numba_cuda-0.0.15/numba_cuda/numba/cuda/tests/cudadrv/test_host_alloc.py +65 -0
  93. numba_cuda-0.0.15/numba_cuda/numba/cuda/tests/cudadrv/test_init.py +139 -0
  94. numba_cuda-0.0.15/numba_cuda/numba/cuda/tests/cudadrv/test_inline_ptx.py +37 -0
  95. numba_cuda-0.0.15/numba_cuda/numba/cuda/tests/cudadrv/test_is_fp16.py +12 -0
  96. numba_cuda-0.0.15/numba_cuda/numba/cuda/tests/cudadrv/test_linker.py +317 -0
  97. numba_cuda-0.0.15/numba_cuda/numba/cuda/tests/cudadrv/test_managed_alloc.py +127 -0
  98. numba_cuda-0.0.15/numba_cuda/numba/cuda/tests/cudadrv/test_mvc.py +54 -0
  99. numba_cuda-0.0.15/numba_cuda/numba/cuda/tests/cudadrv/test_nvvm_driver.py +201 -0
  100. numba_cuda-0.0.15/numba_cuda/numba/cuda/tests/cudadrv/test_pinned.py +37 -0
  101. numba_cuda-0.0.15/numba_cuda/numba/cuda/tests/cudadrv/test_profiler.py +20 -0
  102. numba_cuda-0.0.15/numba_cuda/numba/cuda/tests/cudadrv/test_ptds.py +149 -0
  103. numba_cuda-0.0.15/numba_cuda/numba/cuda/tests/cudadrv/test_reset_device.py +36 -0
  104. numba_cuda-0.0.15/numba_cuda/numba/cuda/tests/cudadrv/test_runtime.py +85 -0
  105. numba_cuda-0.0.15/numba_cuda/numba/cuda/tests/cudadrv/test_select_device.py +41 -0
  106. numba_cuda-0.0.15/numba_cuda/numba/cuda/tests/cudadrv/test_streams.py +122 -0
  107. numba_cuda-0.0.15/numba_cuda/numba/cuda/tests/cudapy/__init__.py +8 -0
  108. numba_cuda-0.0.15/numba_cuda/numba/cuda/tests/cudapy/cache_usecases.py +234 -0
  109. numba_cuda-0.0.15/numba_cuda/numba/cuda/tests/cudapy/cache_with_cpu_usecases.py +41 -0
  110. numba_cuda-0.0.15/numba_cuda/numba/cuda/tests/cudapy/extensions_usecases.py +58 -0
  111. numba_cuda-0.0.15/numba_cuda/numba/cuda/tests/cudapy/jitlink.ptx +30 -0
  112. numba_cuda-0.0.15/numba_cuda/numba/cuda/tests/cudapy/recursion_usecases.py +100 -0
  113. numba_cuda-0.0.15/numba_cuda/numba/cuda/tests/cudapy/test_alignment.py +42 -0
  114. numba_cuda-0.0.15/numba_cuda/numba/cuda/tests/cudapy/test_array.py +260 -0
  115. numba_cuda-0.0.15/numba_cuda/numba/cuda/tests/cudapy/test_array_args.py +201 -0
  116. numba_cuda-0.0.15/numba_cuda/numba/cuda/tests/cudapy/test_array_methods.py +35 -0
  117. numba_cuda-0.0.15/numba_cuda/numba/cuda/tests/cudapy/test_atomics.py +1614 -0
  118. numba_cuda-0.0.15/numba_cuda/numba/cuda/tests/cudapy/test_blackscholes.py +120 -0
  119. numba_cuda-0.0.15/numba_cuda/numba/cuda/tests/cudapy/test_boolean.py +24 -0
  120. numba_cuda-0.0.15/numba_cuda/numba/cuda/tests/cudapy/test_caching.py +545 -0
  121. numba_cuda-0.0.15/numba_cuda/numba/cuda/tests/cudapy/test_casting.py +257 -0
  122. numba_cuda-0.0.15/numba_cuda/numba/cuda/tests/cudapy/test_cffi.py +33 -0
  123. numba_cuda-0.0.15/numba_cuda/numba/cuda/tests/cudapy/test_compiler.py +276 -0
  124. numba_cuda-0.0.15/numba_cuda/numba/cuda/tests/cudapy/test_complex.py +296 -0
  125. numba_cuda-0.0.15/numba_cuda/numba/cuda/tests/cudapy/test_complex_kernel.py +20 -0
  126. numba_cuda-0.0.15/numba_cuda/numba/cuda/tests/cudapy/test_const_string.py +129 -0
  127. numba_cuda-0.0.15/numba_cuda/numba/cuda/tests/cudapy/test_constmem.py +176 -0
  128. numba_cuda-0.0.15/numba_cuda/numba/cuda/tests/cudapy/test_cooperative_groups.py +147 -0
  129. numba_cuda-0.0.15/numba_cuda/numba/cuda/tests/cudapy/test_cuda_array_interface.py +435 -0
  130. numba_cuda-0.0.15/numba_cuda/numba/cuda/tests/cudapy/test_cuda_jit_no_types.py +90 -0
  131. numba_cuda-0.0.15/numba_cuda/numba/cuda/tests/cudapy/test_datetime.py +94 -0
  132. numba_cuda-0.0.15/numba_cuda/numba/cuda/tests/cudapy/test_debug.py +101 -0
  133. numba_cuda-0.0.15/numba_cuda/numba/cuda/tests/cudapy/test_debuginfo.py +221 -0
  134. numba_cuda-0.0.15/numba_cuda/numba/cuda/tests/cudapy/test_device_func.py +222 -0
  135. numba_cuda-0.0.15/numba_cuda/numba/cuda/tests/cudapy/test_dispatcher.py +700 -0
  136. numba_cuda-0.0.15/numba_cuda/numba/cuda/tests/cudapy/test_enums.py +121 -0
  137. numba_cuda-0.0.15/numba_cuda/numba/cuda/tests/cudapy/test_errors.py +79 -0
  138. numba_cuda-0.0.15/numba_cuda/numba/cuda/tests/cudapy/test_exception.py +174 -0
  139. numba_cuda-0.0.15/numba_cuda/numba/cuda/tests/cudapy/test_extending.py +155 -0
  140. numba_cuda-0.0.15/numba_cuda/numba/cuda/tests/cudapy/test_fastmath.py +244 -0
  141. numba_cuda-0.0.15/numba_cuda/numba/cuda/tests/cudapy/test_forall.py +52 -0
  142. numba_cuda-0.0.15/numba_cuda/numba/cuda/tests/cudapy/test_freevar.py +29 -0
  143. numba_cuda-0.0.15/numba_cuda/numba/cuda/tests/cudapy/test_frexp_ldexp.py +66 -0
  144. numba_cuda-0.0.15/numba_cuda/numba/cuda/tests/cudapy/test_globals.py +60 -0
  145. numba_cuda-0.0.15/numba_cuda/numba/cuda/tests/cudapy/test_gufunc.py +456 -0
  146. numba_cuda-0.0.15/numba_cuda/numba/cuda/tests/cudapy/test_gufunc_scalar.py +159 -0
  147. numba_cuda-0.0.15/numba_cuda/numba/cuda/tests/cudapy/test_gufunc_scheduling.py +95 -0
  148. numba_cuda-0.0.15/numba_cuda/numba/cuda/tests/cudapy/test_idiv.py +37 -0
  149. numba_cuda-0.0.15/numba_cuda/numba/cuda/tests/cudapy/test_inspect.py +165 -0
  150. numba_cuda-0.0.15/numba_cuda/numba/cuda/tests/cudapy/test_intrinsics.py +1106 -0
  151. numba_cuda-0.0.15/numba_cuda/numba/cuda/tests/cudapy/test_ipc.py +318 -0
  152. numba_cuda-0.0.15/numba_cuda/numba/cuda/tests/cudapy/test_iterators.py +99 -0
  153. numba_cuda-0.0.15/numba_cuda/numba/cuda/tests/cudapy/test_lang.py +64 -0
  154. numba_cuda-0.0.15/numba_cuda/numba/cuda/tests/cudapy/test_laplace.py +119 -0
  155. numba_cuda-0.0.15/numba_cuda/numba/cuda/tests/cudapy/test_libdevice.py +187 -0
  156. numba_cuda-0.0.15/numba_cuda/numba/cuda/tests/cudapy/test_lineinfo.py +199 -0
  157. numba_cuda-0.0.15/numba_cuda/numba/cuda/tests/cudapy/test_localmem.py +164 -0
  158. numba_cuda-0.0.15/numba_cuda/numba/cuda/tests/cudapy/test_mandel.py +37 -0
  159. numba_cuda-0.0.15/numba_cuda/numba/cuda/tests/cudapy/test_math.py +786 -0
  160. numba_cuda-0.0.15/numba_cuda/numba/cuda/tests/cudapy/test_matmul.py +74 -0
  161. numba_cuda-0.0.15/numba_cuda/numba/cuda/tests/cudapy/test_minmax.py +113 -0
  162. numba_cuda-0.0.15/numba_cuda/numba/cuda/tests/cudapy/test_montecarlo.py +22 -0
  163. numba_cuda-0.0.15/numba_cuda/numba/cuda/tests/cudapy/test_multigpu.py +140 -0
  164. numba_cuda-0.0.15/numba_cuda/numba/cuda/tests/cudapy/test_multiprocessing.py +46 -0
  165. numba_cuda-0.0.15/numba_cuda/numba/cuda/tests/cudapy/test_multithreads.py +101 -0
  166. numba_cuda-0.0.15/numba_cuda/numba/cuda/tests/cudapy/test_nondet.py +49 -0
  167. numba_cuda-0.0.15/numba_cuda/numba/cuda/tests/cudapy/test_operator.py +401 -0
  168. numba_cuda-0.0.15/numba_cuda/numba/cuda/tests/cudapy/test_optimization.py +86 -0
  169. numba_cuda-0.0.15/numba_cuda/numba/cuda/tests/cudapy/test_overload.py +335 -0
  170. numba_cuda-0.0.15/numba_cuda/numba/cuda/tests/cudapy/test_powi.py +124 -0
  171. numba_cuda-0.0.15/numba_cuda/numba/cuda/tests/cudapy/test_print.py +128 -0
  172. numba_cuda-0.0.15/numba_cuda/numba/cuda/tests/cudapy/test_py2_div_issue.py +33 -0
  173. numba_cuda-0.0.15/numba_cuda/numba/cuda/tests/cudapy/test_random.py +104 -0
  174. numba_cuda-0.0.15/numba_cuda/numba/cuda/tests/cudapy/test_record_dtype.py +610 -0
  175. numba_cuda-0.0.15/numba_cuda/numba/cuda/tests/cudapy/test_recursion.py +125 -0
  176. numba_cuda-0.0.15/numba_cuda/numba/cuda/tests/cudapy/test_reduction.py +76 -0
  177. numba_cuda-0.0.15/numba_cuda/numba/cuda/tests/cudapy/test_retrieve_autoconverted_arrays.py +83 -0
  178. numba_cuda-0.0.15/numba_cuda/numba/cuda/tests/cudapy/test_serialize.py +85 -0
  179. numba_cuda-0.0.15/numba_cuda/numba/cuda/tests/cudapy/test_slicing.py +37 -0
  180. numba_cuda-0.0.15/numba_cuda/numba/cuda/tests/cudapy/test_sm.py +444 -0
  181. numba_cuda-0.0.15/numba_cuda/numba/cuda/tests/cudapy/test_sm_creation.py +205 -0
  182. numba_cuda-0.0.15/numba_cuda/numba/cuda/tests/cudapy/test_sync.py +271 -0
  183. numba_cuda-0.0.15/numba_cuda/numba/cuda/tests/cudapy/test_transpose.py +80 -0
  184. numba_cuda-0.0.15/numba_cuda/numba/cuda/tests/cudapy/test_ufuncs.py +277 -0
  185. numba_cuda-0.0.15/numba_cuda/numba/cuda/tests/cudapy/test_userexc.py +47 -0
  186. numba_cuda-0.0.15/numba_cuda/numba/cuda/tests/cudapy/test_vector_type.py +307 -0
  187. numba_cuda-0.0.15/numba_cuda/numba/cuda/tests/cudapy/test_vectorize.py +283 -0
  188. numba_cuda-0.0.15/numba_cuda/numba/cuda/tests/cudapy/test_vectorize_complex.py +20 -0
  189. numba_cuda-0.0.15/numba_cuda/numba/cuda/tests/cudapy/test_vectorize_decor.py +69 -0
  190. numba_cuda-0.0.15/numba_cuda/numba/cuda/tests/cudapy/test_vectorize_device.py +36 -0
  191. numba_cuda-0.0.15/numba_cuda/numba/cuda/tests/cudapy/test_vectorize_scalar_arg.py +37 -0
  192. numba_cuda-0.0.15/numba_cuda/numba/cuda/tests/cudapy/test_warning.py +139 -0
  193. numba_cuda-0.0.15/numba_cuda/numba/cuda/tests/cudapy/test_warp_ops.py +276 -0
  194. numba_cuda-0.0.15/numba_cuda/numba/cuda/tests/cudasim/__init__.py +6 -0
  195. numba_cuda-0.0.15/numba_cuda/numba/cuda/tests/cudasim/support.py +6 -0
  196. numba_cuda-0.0.15/numba_cuda/numba/cuda/tests/cudasim/test_cudasim_issues.py +102 -0
  197. numba_cuda-0.0.15/numba_cuda/numba/cuda/tests/data/__init__.py +0 -0
  198. numba_cuda-0.0.15/numba_cuda/numba/cuda/tests/data/cuda_include.cu +5 -0
  199. numba_cuda-0.0.15/numba_cuda/numba/cuda/tests/data/error.cu +7 -0
  200. numba_cuda-0.0.15/numba_cuda/numba/cuda/tests/data/jitlink.cu +23 -0
  201. numba_cuda-0.0.15/numba_cuda/numba/cuda/tests/data/jitlink.ptx +51 -0
  202. numba_cuda-0.0.15/numba_cuda/numba/cuda/tests/data/warn.cu +7 -0
  203. numba_cuda-0.0.15/numba_cuda/numba/cuda/tests/doc_examples/__init__.py +6 -0
  204. numba_cuda-0.0.15/numba_cuda/numba/cuda/tests/doc_examples/ffi/__init__.py +0 -0
  205. numba_cuda-0.0.15/numba_cuda/numba/cuda/tests/doc_examples/ffi/functions.cu +49 -0
  206. numba_cuda-0.0.15/numba_cuda/numba/cuda/tests/doc_examples/test_cg.py +77 -0
  207. numba_cuda-0.0.15/numba_cuda/numba/cuda/tests/doc_examples/test_cpu_gpu_compat.py +76 -0
  208. numba_cuda-0.0.15/numba_cuda/numba/cuda/tests/doc_examples/test_ffi.py +82 -0
  209. numba_cuda-0.0.15/numba_cuda/numba/cuda/tests/doc_examples/test_laplace.py +155 -0
  210. numba_cuda-0.0.15/numba_cuda/numba/cuda/tests/doc_examples/test_matmul.py +173 -0
  211. numba_cuda-0.0.15/numba_cuda/numba/cuda/tests/doc_examples/test_montecarlo.py +109 -0
  212. numba_cuda-0.0.15/numba_cuda/numba/cuda/tests/doc_examples/test_random.py +59 -0
  213. numba_cuda-0.0.15/numba_cuda/numba/cuda/tests/doc_examples/test_reduction.py +76 -0
  214. numba_cuda-0.0.15/numba_cuda/numba/cuda/tests/doc_examples/test_sessionize.py +130 -0
  215. numba_cuda-0.0.15/numba_cuda/numba/cuda/tests/doc_examples/test_ufunc.py +50 -0
  216. numba_cuda-0.0.15/numba_cuda/numba/cuda/tests/doc_examples/test_vecadd.py +73 -0
  217. numba_cuda-0.0.15/numba_cuda/numba/cuda/tests/nocuda/__init__.py +8 -0
  218. numba_cuda-0.0.15/numba_cuda/numba/cuda/tests/nocuda/test_dummyarray.py +359 -0
  219. numba_cuda-0.0.15/numba_cuda/numba/cuda/tests/nocuda/test_function_resolution.py +36 -0
  220. numba_cuda-0.0.15/numba_cuda/numba/cuda/tests/nocuda/test_import.py +49 -0
  221. numba_cuda-0.0.15/numba_cuda/numba/cuda/tests/nocuda/test_library_lookup.py +238 -0
  222. numba_cuda-0.0.15/numba_cuda/numba/cuda/tests/nocuda/test_nvvm.py +54 -0
  223. numba_cuda-0.0.15/numba_cuda/numba/cuda/types.py +37 -0
  224. numba_cuda-0.0.15/numba_cuda/numba/cuda/ufuncs.py +662 -0
  225. numba_cuda-0.0.15/numba_cuda/numba/cuda/vector_types.py +209 -0
  226. numba_cuda-0.0.15/numba_cuda/numba/cuda/vectorizers.py +252 -0
  227. numba_cuda-0.0.15/numba_cuda.egg-info/PKG-INFO +69 -0
  228. numba_cuda-0.0.15/numba_cuda.egg-info/SOURCES.txt +235 -0
  229. numba_cuda-0.0.15/numba_cuda.egg-info/requires.txt +1 -0
  230. numba_cuda-0.0.15/pyproject.toml +39 -0
  231. numba_cuda-0.0.15/setup.py +70 -0
  232. numba_cuda-0.0.15/site-packages/_numba_cuda_redirector.pth +1 -0
  233. numba_cuda-0.0.15/site-packages/_numba_cuda_redirector.py +76 -0
  234. numba_cuda-0.0.1/PKG-INFO +0 -10
  235. numba_cuda-0.0.1/README.md +0 -3
  236. numba_cuda-0.0.1/numba_cuda.egg-info/PKG-INFO +0 -10
  237. numba_cuda-0.0.1/numba_cuda.egg-info/SOURCES.txt +0 -7
  238. numba_cuda-0.0.1/pyproject.toml +0 -8
  239. {numba_cuda-0.0.1/numba_cuda → numba_cuda-0.0.15/numba_cuda/numba/cuda/kernels}/__init__.py +0 -0
  240. {numba_cuda-0.0.1 → numba_cuda-0.0.15}/numba_cuda.egg-info/dependency_links.txt +0 -0
  241. {numba_cuda-0.0.1 → numba_cuda-0.0.15}/numba_cuda.egg-info/top_level.txt +0 -0
  242. {numba_cuda-0.0.1 → numba_cuda-0.0.15}/setup.cfg +0 -0
@@ -0,0 +1,25 @@
1
+ Copyright (c) 2012, Anaconda, Inc.
2
+ Copyright (c) 2024, NVIDIA CORPORATION.
3
+ All rights reserved.
4
+
5
+ Redistribution and use in source and binary forms, with or without
6
+ modification, are permitted provided that the following conditions are
7
+ met:
8
+
9
+ Redistributions of source code must retain the above copyright notice,
10
+ this list of conditions and the following disclaimer.
11
+
12
+ Redistributions in binary form must reproduce the above copyright
13
+ notice, this list of conditions and the following disclaimer in the
14
+ documentation and/or other materials provided with the distribution.
15
+ THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
16
+ "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
17
+ LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
18
+ A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
19
+ HOLDER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
20
+ SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
21
+ LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
22
+ DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
23
+ THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
24
+ (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
25
+ OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
@@ -0,0 +1,69 @@
1
+ Metadata-Version: 2.1
2
+ Name: numba-cuda
3
+ Version: 0.0.15
4
+ Summary: CUDA target for Numba
5
+ Author: Anaconda Inc., NVIDIA Corporation
6
+ License: BSD 2-clause
7
+ Project-URL: Homepage, https://github.com/rapidsai/numba-cuda
8
+ Project-URL: Documentation, https://github.com/rapidsai/numba-cuda/blob/main/README.md
9
+ Project-URL: Repository, https://github.com/rapidsai/numba-cuda
10
+ Project-URL: License, https://github.com/rapidsai/numba-cuda/blob/main/LICENSE
11
+ Requires-Python: >=3.9
12
+ Description-Content-Type: text/markdown
13
+ License-File: LICENSE
14
+ Requires-Dist: numba>=0.59.1
15
+
16
+ # Numba CUDA Target
17
+
18
+ An out-of-tree CUDA target for Numba.
19
+
20
+ This contains an entire copy of Numba's CUDA target (the `numba.cuda` module),
21
+ and a mechanism to ensure the code from this module (`numba_cuda.numba.cuda`) is
22
+ used as the `numba.cuda` module instead of the code from the `numba` package.
23
+
24
+ This is presently in an early state and is published for testing and feedback.
25
+
26
+ ## Building / testing
27
+
28
+ Install as an editable install:
29
+
30
+ ```
31
+ pip install -e .
32
+ ```
33
+
34
+ Running tests:
35
+
36
+ ```
37
+ python -m numba.runtests numba.cuda.tests
38
+ ```
39
+
40
+ This should discover the`numba.cuda` module from the `numba_cuda` package. You
41
+ can check where `numba.cuda` files are being located by running
42
+
43
+ ```
44
+ python -c "from numba import cuda; print(cuda.__file__)"
45
+ ```
46
+
47
+ which will show a path like:
48
+
49
+ ```
50
+ <path to numba-cuda repo>/numba_cuda/numba/cuda/__init__.py
51
+ ```
52
+
53
+ ## Branching strategy
54
+
55
+ Presently the `main` branch is being used to target the exact behavior of the
56
+ built-in CUDA target. New feature development and bug fixes should be applied to
57
+ `develop`. Once the `main` branch is widely tested and confirmed to work well as
58
+ a drop-in replacement for the built-in `numba.cuda`, the `develop` branch will
59
+ be merged in and new feature development will proceed on `main`.
60
+
61
+ ### Current PR targets
62
+
63
+ - PRs related to replacing the built-in CUDA target's features should target
64
+ `main`.
65
+ - PRs adding new features and bug fixes should target `develop`.
66
+
67
+ ### Future PR targets
68
+
69
+ - In future, all PRs should target the `main` branch.
@@ -0,0 +1,54 @@
1
+ # Numba CUDA Target
2
+
3
+ An out-of-tree CUDA target for Numba.
4
+
5
+ This contains an entire copy of Numba's CUDA target (the `numba.cuda` module),
6
+ and a mechanism to ensure the code from this module (`numba_cuda.numba.cuda`) is
7
+ used as the `numba.cuda` module instead of the code from the `numba` package.
8
+
9
+ This is presently in an early state and is published for testing and feedback.
10
+
11
+ ## Building / testing
12
+
13
+ Install as an editable install:
14
+
15
+ ```
16
+ pip install -e .
17
+ ```
18
+
19
+ Running tests:
20
+
21
+ ```
22
+ python -m numba.runtests numba.cuda.tests
23
+ ```
24
+
25
+ This should discover the`numba.cuda` module from the `numba_cuda` package. You
26
+ can check where `numba.cuda` files are being located by running
27
+
28
+ ```
29
+ python -c "from numba import cuda; print(cuda.__file__)"
30
+ ```
31
+
32
+ which will show a path like:
33
+
34
+ ```
35
+ <path to numba-cuda repo>/numba_cuda/numba/cuda/__init__.py
36
+ ```
37
+
38
+ ## Branching strategy
39
+
40
+ Presently the `main` branch is being used to target the exact behavior of the
41
+ built-in CUDA target. New feature development and bug fixes should be applied to
42
+ `develop`. Once the `main` branch is widely tested and confirmed to work well as
43
+ a drop-in replacement for the built-in `numba.cuda`, the `develop` branch will
44
+ be merged in and new feature development will proceed on `main`.
45
+
46
+ ### Current PR targets
47
+
48
+ - PRs related to replacing the built-in CUDA target's features should target
49
+ `main`.
50
+ - PRs adding new features and bug fixes should target `develop`.
51
+
52
+ ### Future PR targets
53
+
54
+ - In future, all PRs should target the `main` branch.
@@ -0,0 +1 @@
1
+ 0.0.15
@@ -0,0 +1,5 @@
1
+ # Copyright (c) 2024, NVIDIA CORPORATION.
2
+
3
+ from numba_cuda._version import __version__
4
+
5
+ __all__ = ["__version__"]
@@ -0,0 +1,19 @@
1
+ # Copyright (c) 2024, NVIDIA CORPORATION.
2
+ #
3
+ # Licensed under the Apache License, Version 2.0 (the "License");
4
+ # you may not use this file except in compliance with the License.
5
+ # You may obtain a copy of the License at
6
+ #
7
+ # http://www.apache.org/licenses/LICENSE-2.0
8
+ #
9
+ # Unless required by applicable law or agreed to in writing, software
10
+ # distributed under the License is distributed on an "AS IS" BASIS,
11
+ # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
12
+ # See the License for the specific language governing permissions and
13
+ # limitations under the License.
14
+
15
+ import importlib.resources
16
+
17
+ __version__ = (
18
+ importlib.resources.files("numba_cuda").joinpath("VERSION").read_text().strip()
19
+ )
@@ -0,0 +1,22 @@
1
+ from numba import runtests
2
+ from numba.core import config
3
+
4
+ if config.ENABLE_CUDASIM:
5
+ from .simulator_init import *
6
+ else:
7
+ from .device_init import *
8
+ from .device_init import _auto_device
9
+
10
+ from numba.cuda.compiler import (compile, compile_for_current_device,
11
+ compile_ptx, compile_ptx_for_current_device)
12
+
13
+ # This is the out-of-tree NVIDIA-maintained target. This is reported in Numba
14
+ # sysinfo (`numba -s`):
15
+ implementation = "NVIDIA"
16
+
17
+
18
+ def test(*args, **kwargs):
19
+ if not is_available():
20
+ raise cuda_error()
21
+
22
+ return runtests.main("numba.cuda.tests", *args, **kwargs)