File tree Expand file tree Collapse file tree 2 files changed +2
-2
lines changed Expand file tree Collapse file tree 2 files changed +2
-2
lines changed Original file line number Diff line number Diff 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
206206class CPUBackend (BaseBackend ):
Original file line number Diff line number Diff 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
You can’t perform that action at this time.
0 commit comments