Implement new functions from gpurt (GPUOpen-Drivers#2822)

* Implement new functions from gpurt Add AmdTraceRayGetKnownSetRayFlags/AmdTraceRayGetKnownUnsetRayFlags implementations * Add TODO comment
amdrexu · Nov 20, 2023 · 4a90a56 · 4a90a56
1 parent abfdaf5
commit 4a90a56
Show file tree

Hide file tree

Showing 2 changed files with 16 additions and 0 deletions.
diff --git a/llpc/lower/llpcSpirvProcessGpuRtLibrary.cpp b/llpc/lower/llpcSpirvProcessGpuRtLibrary.cpp
@@ -104,6 +104,8 @@ SpirvProcessGpuRtLibrary::LibraryFunctionTable::LibraryFunctionTable() {
   m_libFuncPtrs["AmdTraceRaySetParentId"] = &SpirvProcessGpuRtLibrary::createSetParentId;
   m_libFuncPtrs["AmdTraceRayDispatchRaysIndex"] = &SpirvProcessGpuRtLibrary::createDispatchRayIndex;
   m_libFuncPtrs["AmdTraceRayGetStaticId"] = &SpirvProcessGpuRtLibrary::createGetStaticId;
+  m_libFuncPtrs["AmdTraceRayGetKnownSetRayFlags"] = &SpirvProcessGpuRtLibrary::createGetKnownSetRayFlags;
+  m_libFuncPtrs["AmdTraceRayGetKnownUnsetRayFlags"] = &SpirvProcessGpuRtLibrary::createGetKnownUnsetRayFlags;
 }
 
 // =====================================================================================================================
@@ -639,4 +641,16 @@ void SpirvProcessGpuRtLibrary::createGetStaticId(llvm::Function *func) {
   m_builder->CreateRet(m_builder->create<GpurtGetRayStaticIdOp>());
 }
 
+void SpirvProcessGpuRtLibrary::createGetKnownSetRayFlags(llvm::Function *func) {
+  // TODO: currently return 0 to indicate that there is no known set
+  // We will probably need to analyse the traceRay ray flags for actual value
+  m_builder->CreateRet(m_builder->getInt32(0));
+}
+
+void SpirvProcessGpuRtLibrary::createGetKnownUnsetRayFlags(llvm::Function *func) {
+  // TODO: return 0 to indicate there is no knownUnset bits
+  // We will probably need to analyse the traceRay ray flags for actual value
+  m_builder->CreateRet(m_builder->getInt32(0));
+}
+
 } // namespace Llpc
diff --git a/llpc/lower/llpcSpirvProcessGpuRtLibrary.h b/llpc/lower/llpcSpirvProcessGpuRtLibrary.h
@@ -84,6 +84,8 @@ class SpirvProcessGpuRtLibrary : public SpirvLower, public llvm::PassInfoMixin<S
   void createSetParentId(llvm::Function *func);
   void createDispatchRayIndex(llvm::Function *func);
   void createGetStaticId(llvm::Function *func);
+  void createGetKnownSetRayFlags(llvm::Function *func);
+  void createGetKnownUnsetRayFlags(llvm::Function *func);
   llvm::Value *createGetBvhSrd(llvm::Value *expansion, llvm::Value *boxSortMode);
 };
 } // namespace Llpc