Embed required fallback spv in sycl-post-link

Signed-off-by: jinge90 <[email protected]>

Author: jinge90

llvm/include/llvm/SYCLLowerIR/SYCLRequiredDeviceLibs.h

@@ -40,7 +40,21 @@ enum class DeviceLibExt : std::uint32_t {
   cl_intel_devicelib_bfloat16,
 };
 
+struct SYCLDeviceLibSPVBinary {
+  typedef uint8_t value_type;
+  value_type *SPVRawBytes;
+  size_t SPVBytesNum;
+  SYCLDeviceLibSPVBinary(value_type *RawB, size_t BNum) {
+    SPVRawBytes = RawB;
+    SPVBytesNum = BNum;
+  }
+  value_type *data() const { return SPVRawBytes; }
+  size_t size() const { return SPVBytesNum; }
+};
+
 void getRequiredSYCLDeviceLibs(const Module &M,
-                               SmallVector<StringRef, 16> &ReqLibs);
+                               SmallVector<DeviceLibExt, 16> &ReqLibs);
+
+const char *getDeviceLibFileName(DeviceLibExt RequiredDeviceLibExt);
 
 } // namespace llvm

llvm/lib/SYCLLowerIR/ComputeModuleRuntimeInfo.cpp

@@ -21,6 +21,8 @@
 #include "llvm/SYCLLowerIR/SYCLRequiredDeviceLibs.h"
 #include "llvm/SYCLLowerIR/SYCLUtils.h"
 #include "llvm/SYCLLowerIR/SpecConstants.h"
+#include "llvm/Support/FileSystem.h"
+#include "llvm/Support/MemoryBuffer.h"
 #include <queue>
 #include <unordered_set>
 
@@ -167,9 +169,29 @@ PropSetRegTy computeModuleProperties(const Module &M,
 
   PropSetRegTy PropSet;
   {
-    SmallVector<StringRef, 16> RequiredLibs;
+    SmallVector<llvm::DeviceLibExt, 16> RequiredLibs;
     llvm::getRequiredSYCLDeviceLibs(M, RequiredLibs);
-    for (auto RL : RequiredLibs) {
+    for (auto Ext : RequiredLibs) {
+      const char *SPVFileName = llvm::getDeviceLibFileName(Ext);
+      std::string SPVPath =
+          DeviceLibSPVLoc.str() + "/" + std::string(SPVFileName);
+      if (!llvm::sys::fs::exists(SPVPath))
+        continue;
+
+      auto SPVMB = llvm::MemoryBuffer::getFile(SPVPath);
+      if (!SPVMB)
+        continue;
+
+      size_t SPVSize = (*SPVMB)->getBufferSize();
+      uint8_t *SPVBuffer = reinterpret_cast<uint8_t *>(
+          std::aligned_alloc(alignof(uint32_t), SPVSize + sizeof(uint32_t)));
+      *(reinterpret_cast<uint32_t *>(SPVBuffer)) = static_cast<uint32_t>(Ext);
+      std::memcpy(SPVBuffer + 1, (*SPVMB)->getBufferStart(), SPVSize);
+      llvm::SYCLDeviceLibSPVBinary SPVBinaryObj(SPVBuffer,
+                                                SPVSize + sizeof(uint32_t));
+      PropSet.add(PropSetRegTy::SYCL_DEVICELIB_REQ_BINS, SPVFileName,
+                  SPVBinaryObj);
+      std::free(SPVBuffer);
     }
   }
 

llvm/lib/SYCLLowerIR/SYCLRequiredDeviceLibs.cpp

@@ -760,7 +760,7 @@ namespace llvm {
 // 2. The function is declaration which means it doesn't have function body
 // And we don't expect non-spirv functions with "__devicelib_" prefix.
 void getRequiredSYCLDeviceLibs(
-    const Module &M, llvm::SmallVector<llvm::StringRef, 16> &ReqDeviceLibs) {
+    const Module &M, llvm::SmallVector<DeviceLibExt, 16> &ReqDeviceLibs) {
   // Device libraries will be enabled only for spir-v module.
   if (!Triple(M.getTargetTriple()).isSPIROrSPIRV())
     return;
@@ -776,9 +776,12 @@ void getRequiredSYCLDeviceLibs(
         continue;
 
       DeviceLibUsed.insert(DeviceLibFuncIter->second);
-      ReqDeviceLibs.push_back(DeviceLibSPVExtMap[DeviceLibFuncIter->second]);
+      ReqDeviceLibs.push_back(DeviceLibFuncIter->second);
     }
   }
 }
 
+const char *getDeviceLibFileName(DeviceLibExt RequiredDeviceLibExt) {
+  return DeviceLibSPVExtMap[RequiredDeviceLibExt];
+}
 } // namespace llvm