Skip to content

Commit 15b3f87

Browse files
authored
Use sha256 to satisfy weak hash query. (#330)
CodeQL flags usage of md5 so switch to sha256 like used in the rest of Triton.
1 parent eaeb554 commit 15b3f87

File tree

2 files changed

+2
-2
lines changed

2 files changed

+2
-2
lines changed

backend/compiler.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -200,7 +200,7 @@ def __post_init__(self):
200200

201201
def hash(self):
202202
key = '_'.join([f'{name}-{val}' for name, val in self.__dict__.items()])
203-
return hashlib.md5(key.encode("utf-8")).hexdigest()
203+
return hashlib.sha256(key.encode("utf-8")).hexdigest()
204204

205205

206206
class CPUBackend(BaseBackend):

backend/driver.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -269,7 +269,7 @@ def launch(
269269
kernel_name = kernel_metadata[6] # see pack_metadata in compiler.py
270270
src = launcher_src.replace(kernel_placeholder_name, kernel_name)
271271

272-
key = hashlib.md5(src.encode("utf-8") + kernel_obj).hexdigest()
272+
key = hashlib.sha256(src.encode("utf-8") + kernel_obj).hexdigest()
273273
cache = get_cache_manager(key)
274274
name = "__triton_shared_ref_cpu_kernel_launcher"
275275

0 commit comments

Comments
 (0)