ó \ K]c@sÁdZddlZddlmZddlmZmZeejƒddd„ƒYƒZdefd„ƒYZ d efd „ƒYZ d efd „ƒYZ e d „Z dd d ddgZ dS(sw Hints to wrap Kernel arguments to indicate how to manage host-device memory transfers before & after the kernel call. iÿÿÿÿN(t add_metaclass(ttypeoftPurposetArgHintcBs5eZd„Zejdd„ƒZed„ƒZRS(cCs ||_dS(N(tvalue(tselfR((s.lib/python2.7/site-packages/numba/cuda/args.pyt__init__sicCsdS(s# :param stream: a stream to use when copying data :param retr: a list of clean-up work to do after the kernel's been run. Append 0-arg lambdas to it! :return: a value (usually an `DeviceNDArray`) to be passed to the kernel N((Rtretrtstream((s.lib/python2.7/site-packages/numba/cuda/args.pyt to_devices cCst|jtjƒS(N(RRRtargument(R((s.lib/python2.7/site-packages/numba/cuda/args.pyt _numba_type_s(t__name__t __module__RtabctabstractmethodR tpropertyR (((s.lib/python2.7/site-packages/numba/cuda/args.pyR s  tIncBseZdd„ZRS(icsEddlm}||jd|ƒ\‰}|j‡fd†ƒˆS(Ni(t auto_deviceRcsˆS(N(((tdevary(s.lib/python2.7/site-packages/numba/cuda/args.pyt*t(tcudadrv.devicearrayRRtappend(RRRRt_((Rs.lib/python2.7/site-packages/numba/cuda/args.pyR #s  (R R R (((s.lib/python2.7/site-packages/numba/cuda/args.pyR"stOutcBseZdd„ZRS(icsZddlm}|ˆjdtdˆƒ\‰}|rV|j‡‡‡fd†ƒnˆS(Ni(RtcopyRcsˆjˆjdˆƒS(NR(t copy_to_hostR((RRR(s.lib/python2.7/site-packages/numba/cuda/args.pyR6R(RRRtFalseR(RRRRtconv((RRRs.lib/python2.7/site-packages/numba/cuda/args.pyR /s (R R R (((s.lib/python2.7/site-packages/numba/cuda/args.pyR.stInOutcBseZdd„ZRS(icsTddlm}|ˆjdˆƒ\‰}|rP|j‡‡‡fd†ƒnˆS(Ni(RRcsˆjˆjdˆƒS(NR(RR((RRR(s.lib/python2.7/site-packages/numba/cuda/args.pyRAR(RRRR(RRRRR((RRRs.lib/python2.7/site-packages/numba/cuda/args.pyR ;s (R R R (((s.lib/python2.7/site-packages/numba/cuda/args.pyR:scCst|tƒr|S||ƒS(N(t isinstanceR(Rtdefault((s.lib/python2.7/site-packages/numba/cuda/args.pytwrap_argEsR!((t__doc__Rt numba.sixRtnumba.typing.typeofRRtABCMetaRRRRR!t__all__(((s.lib/python2.7/site-packages/numba/cuda/args.pyts "