third_party: ngen: add movi src check

oneapi-src · Jan 23, 2025 · f3ceb3d · f3ceb3d
1 parent 0cdbbe9
commit f3ceb3d
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 0 deletions.
diff --git a/third_party/ngen/ngen.hpp b/third_party/ngen/ngen.hpp
@@ -1947,6 +1947,9 @@ BinaryCodeGenerator<hw>::opX(Opcode op, DataType defaultType, const InstructionM
 
     typename EncodingTag12Dispatch<hw>::tag tag;
     Instruction12 i{};
+#ifdef NGEN_SAFE
+    if (!src0.isIndirect())  throw invalid_address_mode_exception();
+#endif
 
     InstructionModifier emod = mod | defaultModifier;
     if (forceWE)

diff --git a/third_party/ngen/ngen_core.hpp b/third_party/ngen/ngen_core.hpp
@@ -214,6 +214,10 @@ class invalid_address_modifier_exception : public std::runtime_error {
 public:
     invalid_address_modifier_exception() : std::runtime_error("Invalid address offset") {}
 };
+class invalid_address_mode_exception : public std::runtime_error {
+public:
+    invalid_address_mode_exception() : std::runtime_error("Invalid address mode") {}
+};
 #endif
 
 // Graphics core generations.