ó Ä 9j+ãóЕSrSSKJr SSKrSSKrSSKrSSKrS Sjr\"\"S55r \RRS5r "SS\R5r"S S \5rg) a“Kernel cache backend infrastructure for CuPy compiler. This module provides cache backend abstractions for storing compiled kernel binaries. The cache helps avoid recompiling kernels that have already been compiled. .. warning:: User-defined custom cache backends are experimental and should not expect API stability. The interface may change in future releases without deprecation warnings. é)Ú annotationsNcóH•[R"USS9R5$)z%Compute SHA1 hash of the given bytes.F)Úusedforsecurity)ÚhashlibÚsha1Ú hexdigest)Úvalues ÚS/home/wildlama/miniconda3/lib/python3.13/site-packages/cupy/cuda/_compiler_cache.pyÚ_hash_hexdigestr s€ä <Š<˜¨uÑ 5× ?Ñ ?Ó AÐAór z~/.cupy/kernel_cachecóh•\rSrSrSr\R SSj5r\R SSj5rSr g) ÚKernelCacheBackendéaˆAbstract base class for kernel cache storage backends. This class defines the interface for pluggable cache storage backends. Subclasses should implement methods to load and save compiled kernel binaries. .. warning:: User-defined custom cache backends are experimental and should not expect API stability. The interface may change in future releases without deprecation warnings. có•[e)zÛLoad a cached kernel binary. Args: name (str): The cache key (filename) for the compiled kernel. Returns: bytes or None: The cubin binary data (without hash prefix) if found and valid, None otherwise. ©ÚNotImplementedError)ÚselfÚnames r ÚloadÚKernelCacheBackend.load+s €ô"Ð!r có•[e)aSave a compiled kernel binary to cache. This method may perform I/O asynchronously to avoid blocking kernel execution. Args: name (str): The cache key (filename) for the compiled kernel. cubin (bytes): The compiled kernel binary data. source (str): The CUDA source code. r)rrÚcubinÚsources r ÚsaveÚKernelCacheBackend.save8s €ô"Ð!r ©N©rÚstrÚreturnú bytes | None©rrrÚbytesrrrÚNone) Ú__name__Ú __module__Ú __qualname__Ú__firstlineno__Ú__doc__ÚabcÚabstractmethodrrÚ__static_attributes__rr r rrs;†ñ ð ×Ñó "óð "ð ×Ñó "óó "r rcóX•\rSrSrSrS S SjjrS SjrSSjrSSjrSSjr SS jr S r g)ÚDiskKernelCacheBackendéGzoDisk-based kernel cache storage backend. This backend stores compiled kernel binaries in a directory on disk. NcóR•Uc$[RRS[5nXl[R R UR5(d[R"URSS9 [[RRS55Ul g)zÂInitialize the disk cache backend. Args: cache_dir (str, optional): Directory to store cache files. Defaults to CUPY_CACHE_DIR environment variable or ~/.cupy/kernel_cache. NÚCUPY_CACHE_DIRT)Úexist_okÚCUPY_CACHE_SAVE_CUDA_SOURCE) ÚosÚenvironÚgetÚ_default_cache_dirÚ _cache_dirÚpathÚisdirÚmakedirsÚboolÚ_save_cuda_source)rÚ cache_dirs r Ú__init__ÚDiskKernelCacheBackend.__init__Msl€ð Ñ ÜŸ ™ Ÿ™Ð'7Ô9KÓLˆIØ#ŒÜw‰w}‰}˜TŸ_™_×-Ñ-Ü KŠK˜Ÿ™°$Ò 7Ü!%Ü J‰JN‰NÐ8Ó 9ó";ˆÕr có<•[U5RS5U-$)zÂEncode a cubin binary to the on-disk format (hash prefix + cubin). Args: cubin (bytes): Raw compiled kernel binary data. Returns: bytes: SHA-1 hash (ASCII hex) prepended to the cubin. Úascii)r Úencode)rrs r Ú _encode_cubinÚ$DiskKernelCacheBackend._encode_cubin]s€ô˜uÓ%×,Ñ,¨WÓ5¸Ñ=Ð=r có•[U5[:agUS[nU[SnU[U5RS5:wagU$)zÊDecode and validate data in the on-disk format. Args: data (bytes): Raw bytes in the on-disk format (hash + cubin). Returns: bytes or None: The raw cubin if the hash is valid, None otherwise. NrA)ÚlenÚ _hash_lengthr rB)rÚdataÚ hash_storedrs r Ú _decode_cubinÚ$DiskKernelCacheBackend._decode_cubinhsK€ô ˆt‹9”|Ó #Øؘ=œLÐ)ˆ Ø”\]Ð#ˆØ œ/¨%Ó0×7Ñ7¸Ó@Ó @Ø؈ r cóT•[RRURU5n[R "URSS9nUR U5 URnSSS5 [R"WU5 g!,(df  N'=f![a gf=f)auAtomically write pre-encoded (hash + cubin) data to the cache dir. Unlike :meth:`save`, this method accepts data that is already in the on-disk format (i.e. the SHA-1 hash prefix is already prepended). It does not save a `.cu` source file. Args: name (str): The cache key (filename) for the compiled kernel. data (bytes): Pre-encoded bytes (hash prefix + cubin). F)ÚdirÚdeleteN) r3r8Újoinr7ÚtempfileÚNamedTemporaryFileÚwriterÚreplaceÚPermissionError)rrrHr8ÚtfÚ temp_paths r Ú_write_encodedÚ%DiskKernelCacheBackend._write_encodedys€€ôw‰w|‰|˜DŸO™O¨TÓ2ˆÜ × (Ò (Ø—O‘O¨Eò3Ø68Ø H‰HTŒNØŸ™ˆI÷3ð Ü JŠJy $Õ '÷ 3õ3ûô ó Ù ð úsÁ B Á1B B B'Â&B'có.•[RRURU5n[RR U5(dg[ US5nUR 5nSSS5 URW5$!,(df  N=f)zåLoad a cached kernel binary from disk. Args: name (str): The cache key (filename) for the compiled kernel. Returns: bytes or None: The cubin binary data (without hash prefix) if found and valid, None otherwise. NÚrb)r3r8rOr7ÚexistsÚopenÚreadrJ)rrr8ÚfilerHs r rÚDiskKernelCacheBackend.loadŽsj€ôw‰w|‰|˜DŸO™O¨TÓ2ˆÜw‰w~‰~˜d×#Ñ#Øä $˜Ô  Ø—9‘9“;ˆD÷ð×!Ñ! $Ó'Ð'÷Õ ús ÁB Bcó0•URXRU55 UR(aT[RR UR U5n[US-S5nURU5 SSS5 gg!,(df  g=f)zÌSave a compiled kernel binary to disk. Args: name (str): The cache key (filename) for the compiled kernel. cubin (bytes): The compiled kernel binary data. source (str): The CUDA source code. z.cuÚwN) rWrCr<r3r8rOr7r\rR)rrrrr8Úfs r rÚDiskKernelCacheBackend.save¡so€ð ×јD×"4Ñ"4°UÓ";Ô<ð × !× !Ü—7‘7—<‘< §¡°Ó6ˆDÜd˜U‘l CÔ(¨AØ—‘˜”÷)Ð(ð "ç(Õ(ús Á+B B)r7r<)N)r=z str | None)rr"rr")rHr"rr )rrrHr"rr#rr!) r$r%r&r'r(r>rCrJrWrrr+rr r r-r-Gs%†ñö ;ô >ôô"ô*(÷& r r-)r r"rr)r(Ú __future__rr)rr3rPr rFrGr8Ú expanduserr6ÚABCrr-rr r Úrgshðñ õ#ã ÛÛ ÛôBñ ‘? 3Ó'Ó(€ Ø—W‘W×'Ñ'Ð(>Ó?Ðô&"˜Ÿ™ô&"ôRh Ð/õh r