WIP

lib/JitEngineDevice.hpp

@@ -78,6 +78,7 @@ class JITKernelInfo {
   const auto &getRCIndices() const { return RCIndices; }
   const auto &getRCTypes() const { return RCTypes; }
   const auto &getNumRCs() const { return NumRCs; }
+  const auto &getSHA256() const { return SHA256; }
 };
 
 struct FatbinWrapper_t {
@@ -545,7 +546,7 @@ JitEngineDevice<ImplT>::compileAndRun(
                    : extractDeviceBitcode(KernelName, Kernel);
 
   // I have already read the LLVM IR from the Binary. Pick the Static Hash
-  auto StaticHash = SHA256HashWithBitcodes[Index].first;
+  auto StaticHash = JITKernelInfoMap[Kernel].getSHA256();
   // TODO: This does not include the GridDims/BlockDims. We need to fix it.
   auto PersistentHash = ProteusDeviceBinHash;
   uint64_t DynamicHashValue =
@@ -554,7 +555,7 @@ JitEngineDevice<ImplT>::compileAndRun(
   KernelFunc = CodeCache.lookup(DynamicHashValue);
   std::cout << " Function with name " << KernelName.str() << "at address "
             << Kernel << " has PersistentHash " << PersistentHash
-            << " Static Hash:" << StaticHash
+            << " Static Hash:" << StaticHash.str()
             << " Dynamic Hash:" << DynamicHashValue << "\n";
 
   // We found the kernel, execute