@@ -68,9 +68,7 @@ def _extract_bytes(value: object) -> bytes:
6868 if isinstance (code , str ):
6969 return Path (code ).read_bytes ()
7070 return bytes (code )
71- raise TypeError (
72- f"cache values must be bytes-like or ObjectCode, got { type (value ).__name__ } "
73- )
71+ raise TypeError (f"cache values must be bytes-like or ObjectCode, got { type (value ).__name__ } " )
7472
7573
7674def _as_key_bytes (key : object ) -> bytes :
@@ -117,8 +115,11 @@ class ProgramCacheResource(abc.ABC):
117115 from cuda.core._module import ObjectCode
118116
119117 key = make_program_cache_key(
120- code=source, code_type="c++", options=options,
121- target_type="cubin", extra_digest=header_fingerprint(),
118+ code=source,
119+ code_type="c++",
120+ options=options,
121+ target_type="cubin",
122+ extra_digest=header_fingerprint(),
122123 )
123124 data = cache.get(key)
124125 if data is None:
@@ -147,9 +148,7 @@ def __getitem__(self, key: bytes | str) -> bytes:
147148 """
148149
149150 @abc .abstractmethod
150- def __setitem__ (
151- self , key : bytes | str , value : bytes | bytearray | memoryview | ObjectCode
152- ) -> None :
151+ def __setitem__ (self , key : bytes | str , value : bytes | bytearray | memoryview | ObjectCode ) -> None :
153152 """Store ``value`` under ``key``.
154153
155154 Path-backed :class:`~cuda.core.ObjectCode` is read from disk at
@@ -189,9 +188,7 @@ def update(
189188 self ,
190189 items : (
191190 collections .abc .Mapping [bytes | str , bytes | bytearray | memoryview | ObjectCode ]
192- | collections .abc .Iterable [
193- tuple [bytes | str , bytes | bytearray | memoryview | ObjectCode ]
194- ]
191+ | collections .abc .Iterable [tuple [bytes | str , bytes | bytearray | memoryview | ObjectCode ]]
195192 ),
196193 / ,
197194 ) -> None :
@@ -567,9 +564,7 @@ def make_program_cache_key(
567564 options = ProgramOptions(arch="sm_80")
568565
569566 with FileStreamProgramCache() as cache:
570- obj = Program(source, "c++", options=options).compile(
571- "cubin", cache=cache
572- )
567+ obj = Program(source, "c++", options=options).compile("cubin", cache=cache)
573568
574569 Call ``make_program_cache_key`` directly when the compile inputs
575570 require an ``extra_digest`` (the cache cannot read external file
@@ -940,9 +935,7 @@ def __getitem__(self, key: object) -> bytes:
940935 self ._entries .move_to_end (k )
941936 return data
942937
943- def __setitem__ (
944- self , key : object , value : bytes | bytearray | memoryview | ObjectCode
945- ) -> None :
938+ def __setitem__ (self , key : object , value : bytes | bytearray | memoryview | ObjectCode ) -> None :
946939 data = _extract_bytes (value )
947940 size = len (data )
948941 k = _as_key_bytes (key )
@@ -1367,9 +1360,7 @@ def __getitem__(self, key: object) -> bytes:
13671360 _touch_atime (path , st )
13681361 return data
13691362
1370- def __setitem__ (
1371- self , key : object , value : bytes | bytearray | memoryview | ObjectCode
1372- ) -> None :
1363+ def __setitem__ (self , key : object , value : bytes | bytearray | memoryview | ObjectCode ) -> None :
13731364 data = _extract_bytes (value )
13741365 target = self ._path_for_key (key )
13751366 target .parent .mkdir (parents = True , exist_ok = True )
0 commit comments