__init__.pyi 5.3 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249
  1. import typing
  2. # Enumerations
  3. OCL_VECTOR_OWN: int
  4. OCL_VECTOR_MAX: int
  5. OCL_VECTOR_DEFAULT: int
  6. OclVectorStrategy = int
  7. """One of [OCL_VECTOR_OWN, OCL_VECTOR_MAX, OCL_VECTOR_DEFAULT]"""
  8. Device_TYPE_DEFAULT: int
  9. DEVICE_TYPE_DEFAULT: int
  10. Device_TYPE_CPU: int
  11. DEVICE_TYPE_CPU: int
  12. Device_TYPE_GPU: int
  13. DEVICE_TYPE_GPU: int
  14. Device_TYPE_ACCELERATOR: int
  15. DEVICE_TYPE_ACCELERATOR: int
  16. Device_TYPE_DGPU: int
  17. DEVICE_TYPE_DGPU: int
  18. Device_TYPE_IGPU: int
  19. DEVICE_TYPE_IGPU: int
  20. Device_TYPE_ALL: int
  21. DEVICE_TYPE_ALL: int
  22. Device_FP_DENORM: int
  23. DEVICE_FP_DENORM: int
  24. Device_FP_INF_NAN: int
  25. DEVICE_FP_INF_NAN: int
  26. Device_FP_ROUND_TO_NEAREST: int
  27. DEVICE_FP_ROUND_TO_NEAREST: int
  28. Device_FP_ROUND_TO_ZERO: int
  29. DEVICE_FP_ROUND_TO_ZERO: int
  30. Device_FP_ROUND_TO_INF: int
  31. DEVICE_FP_ROUND_TO_INF: int
  32. Device_FP_FMA: int
  33. DEVICE_FP_FMA: int
  34. Device_FP_SOFT_FLOAT: int
  35. DEVICE_FP_SOFT_FLOAT: int
  36. Device_FP_CORRECTLY_ROUNDED_DIVIDE_SQRT: int
  37. DEVICE_FP_CORRECTLY_ROUNDED_DIVIDE_SQRT: int
  38. Device_EXEC_KERNEL: int
  39. DEVICE_EXEC_KERNEL: int
  40. Device_EXEC_NATIVE_KERNEL: int
  41. DEVICE_EXEC_NATIVE_KERNEL: int
  42. Device_NO_CACHE: int
  43. DEVICE_NO_CACHE: int
  44. Device_READ_ONLY_CACHE: int
  45. DEVICE_READ_ONLY_CACHE: int
  46. Device_READ_WRITE_CACHE: int
  47. DEVICE_READ_WRITE_CACHE: int
  48. Device_NO_LOCAL_MEM: int
  49. DEVICE_NO_LOCAL_MEM: int
  50. Device_LOCAL_IS_LOCAL: int
  51. DEVICE_LOCAL_IS_LOCAL: int
  52. Device_LOCAL_IS_GLOBAL: int
  53. DEVICE_LOCAL_IS_GLOBAL: int
  54. Device_UNKNOWN_VENDOR: int
  55. DEVICE_UNKNOWN_VENDOR: int
  56. Device_VENDOR_AMD: int
  57. DEVICE_VENDOR_AMD: int
  58. Device_VENDOR_INTEL: int
  59. DEVICE_VENDOR_INTEL: int
  60. Device_VENDOR_NVIDIA: int
  61. DEVICE_VENDOR_NVIDIA: int
  62. KernelArg_LOCAL: int
  63. KERNEL_ARG_LOCAL: int
  64. KernelArg_READ_ONLY: int
  65. KERNEL_ARG_READ_ONLY: int
  66. KernelArg_WRITE_ONLY: int
  67. KERNEL_ARG_WRITE_ONLY: int
  68. KernelArg_READ_WRITE: int
  69. KERNEL_ARG_READ_WRITE: int
  70. KernelArg_CONSTANT: int
  71. KERNEL_ARG_CONSTANT: int
  72. KernelArg_PTR_ONLY: int
  73. KERNEL_ARG_PTR_ONLY: int
  74. KernelArg_NO_SIZE: int
  75. KERNEL_ARG_NO_SIZE: int
  76. # Classes
  77. class Device:
  78. # Functions
  79. def __init__(self) -> None: ...
  80. def name(self) -> str: ...
  81. def extensions(self) -> str: ...
  82. def isExtensionSupported(self, extensionName: str) -> bool: ...
  83. def version(self) -> str: ...
  84. def vendorName(self) -> str: ...
  85. def OpenCL_C_Version(self) -> str: ...
  86. def OpenCLVersion(self) -> str: ...
  87. def deviceVersionMajor(self) -> int: ...
  88. def deviceVersionMinor(self) -> int: ...
  89. def driverVersion(self) -> str: ...
  90. def type(self) -> int: ...
  91. def addressBits(self) -> int: ...
  92. def available(self) -> bool: ...
  93. def compilerAvailable(self) -> bool: ...
  94. def linkerAvailable(self) -> bool: ...
  95. def doubleFPConfig(self) -> int: ...
  96. def singleFPConfig(self) -> int: ...
  97. def halfFPConfig(self) -> int: ...
  98. def endianLittle(self) -> bool: ...
  99. def errorCorrectionSupport(self) -> bool: ...
  100. def executionCapabilities(self) -> int: ...
  101. def globalMemCacheSize(self) -> int: ...
  102. def globalMemCacheType(self) -> int: ...
  103. def globalMemCacheLineSize(self) -> int: ...
  104. def globalMemSize(self) -> int: ...
  105. def localMemSize(self) -> int: ...
  106. def localMemType(self) -> int: ...
  107. def hostUnifiedMemory(self) -> bool: ...
  108. def imageSupport(self) -> bool: ...
  109. def imageFromBufferSupport(self) -> bool: ...
  110. def intelSubgroupsSupport(self) -> bool: ...
  111. def image2DMaxWidth(self) -> int: ...
  112. def image2DMaxHeight(self) -> int: ...
  113. def image3DMaxWidth(self) -> int: ...
  114. def image3DMaxHeight(self) -> int: ...
  115. def image3DMaxDepth(self) -> int: ...
  116. def imageMaxBufferSize(self) -> int: ...
  117. def imageMaxArraySize(self) -> int: ...
  118. def vendorID(self) -> int: ...
  119. def isAMD(self) -> bool: ...
  120. def isIntel(self) -> bool: ...
  121. def isNVidia(self) -> bool: ...
  122. def maxClockFrequency(self) -> int: ...
  123. def maxComputeUnits(self) -> int: ...
  124. def maxConstantArgs(self) -> int: ...
  125. def maxConstantBufferSize(self) -> int: ...
  126. def maxMemAllocSize(self) -> int: ...
  127. def maxParameterSize(self) -> int: ...
  128. def maxReadImageArgs(self) -> int: ...
  129. def maxWriteImageArgs(self) -> int: ...
  130. def maxSamplers(self) -> int: ...
  131. def maxWorkGroupSize(self) -> int: ...
  132. def maxWorkItemDims(self) -> int: ...
  133. def memBaseAddrAlign(self) -> int: ...
  134. def nativeVectorWidthChar(self) -> int: ...
  135. def nativeVectorWidthShort(self) -> int: ...
  136. def nativeVectorWidthInt(self) -> int: ...
  137. def nativeVectorWidthLong(self) -> int: ...
  138. def nativeVectorWidthFloat(self) -> int: ...
  139. def nativeVectorWidthDouble(self) -> int: ...
  140. def nativeVectorWidthHalf(self) -> int: ...
  141. def preferredVectorWidthChar(self) -> int: ...
  142. def preferredVectorWidthShort(self) -> int: ...
  143. def preferredVectorWidthInt(self) -> int: ...
  144. def preferredVectorWidthLong(self) -> int: ...
  145. def preferredVectorWidthFloat(self) -> int: ...
  146. def preferredVectorWidthDouble(self) -> int: ...
  147. def preferredVectorWidthHalf(self) -> int: ...
  148. def printfBufferSize(self) -> int: ...
  149. def profilingTimerResolution(self) -> int: ...
  150. @classmethod
  151. def getDefault(cls) -> Device: ...
  152. class OpenCLExecutionContext:
  153. ...
  154. # Functions
  155. def finish() -> None: ...
  156. def haveAmdBlas() -> bool: ...
  157. def haveAmdFft() -> bool: ...
  158. def haveOpenCL() -> bool: ...
  159. def setUseOpenCL(flag: bool) -> None: ...
  160. def useOpenCL() -> bool: ...