fix to_qir function

Signed-off-by: Alex McCaskey <amccaskey@nvidia.com>
NVIDIA · Mar 25, 2024 · 8764dc2 · 8764dc2
1 parent c1d65f3
commit 8764dc2
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 12 deletions.
diff --git a/python/runtime/cudaq/platform/py_alt_launch_kernel.cpp b/python/runtime/cudaq/platform/py_alt_launch_kernel.cpp
@@ -375,8 +375,9 @@ void bindAltLaunchKernel(py::module &mod) {
         auto name = kernel.attr("name").cast<std::string>();
         auto kernelFuncOp = getKernelFuncOp(module, name);
         cudaq::OpaqueArguments args;
-        cudaq::packArgs(args, runtimeArgs, kernelFuncOp,
-                        [](OpaqueArguments &, py::object &) { return false; });
+        cudaq::packArgs(
+            args, runtimeArgs, kernelFuncOp,
+            [](OpaqueArguments &, py::object &) { return false; }, false);
         return getQIRLL(name, module, args, profile);
       },
       py::arg("kernel"), py::kw_only(), py::arg("profile") = "");

diff --git a/python/utils/OpaqueArguments.h b/python/utils/OpaqueArguments.h
@@ -184,16 +184,17 @@ inline mlir::func::FuncOp getKernelFuncOp(MlirModule module,
   return kernelFunc;
 }
 
-inline void
-packArgs(OpaqueArguments &argData, py::args args,
-         mlir::func::FuncOp kernelFuncOp,
-         const std::function<bool(OpaqueArguments &argData, py::object &arg)>
-             &backupHandler) {
-  if (kernelFuncOp.getNumArguments() != args.size())
-    throw std::runtime_error("Invalid runtime arguments - kernel expected " +
-                             std::to_string(kernelFuncOp.getNumArguments()) +
-                             " but was provided " +
-                             std::to_string(args.size()) + " arguments.");
+inline void packArgs(OpaqueArguments &argData, py::args args,
+                     mlir::func::FuncOp kernelFuncOp,
+                     const std::function<bool(OpaqueArguments &argData,
+                                              py::object &arg)> &backupHandler,
+                     bool checkNumArgs = false) {
+  if (checkNumArgs)
+    if (kernelFuncOp.getNumArguments() != args.size())
+      throw std::runtime_error("Invalid runtime arguments - kernel expected " +
+                               std::to_string(kernelFuncOp.getNumArguments()) +
+                               " but was provided " +
+                               std::to_string(args.size()) + " arguments.");
 
   for (std::size_t i = 0; i < args.size(); i++) {
     py::object arg = args[i];