Make hashing work for tensorflow_jit

src/pystencils_autodiff/tensorflow_jit.py

@@ -7,6 +7,7 @@
 """
 
 """
+import hashlib
 import json
 import subprocess
 import sysconfig
@@ -18,6 +19,9 @@ import pystencils
 from pystencils.cpu.cpujit import get_cache_config, get_compiler_config, get_pystencils_include_path
 from pystencils_autodiff._file_io import read_file, write_file
 
+_hash = hashlib.md5
+
+
 # TODO: msvc
 if get_compiler_config()['os'] != 'windows':
     _shared_object_flag = '-shared'
@@ -56,7 +60,8 @@ def link_and_load(object_files, destination_file=None, overwrite_destination_fil
     """
 
     if not destination_file:
-        destination_file = join(get_cache_config()['object_cache'], f"{abs(hash(tuple(object_files))):x}.so")
+        destination_file = join(get_cache_config()['object_cache'],
+                                f"{_hash('.'.join(sorted(object_files)).encode()).hexdigest()}.so")
 
     if not exists(destination_file) or overwrite_destination_file:
         command = [get_compiler_config()['command'],
@@ -136,6 +141,7 @@ def compile_file(file, use_nvcc=False, nvcc='nvcc', overwrite_destination_file=T
 
     if not exists(destination_file) or overwrite_destination_file:
         subprocess.check_call(command)
+
     return destination_file
 
 
@@ -152,7 +158,7 @@ def compile_sources_and_load(host_sources, cuda_sources=[], additional_compile_f
             source_code = source
 
         file_extension = '.cu' if is_cuda else '.cpp'
-        file_name = join(pystencils.cache.cache_dir, f'{abs(hash(source_code)):x}{file_extension}')
+        file_name = join(pystencils.cache.cache_dir, f'{_hash(source_code.encode()).hexdigest()}{file_extension}')
         write_file(file_name, source_code)
 
         compile_file(file_name, use_nvcc=is_cuda, overwrite_destination_file=False,