Add execution mode option to PyHDK python tests

omniscidb/ConfigBuilder/ConfigBuilder.cpp

@@ -574,6 +574,12 @@ bool ConfigBuilder::parseCommandLineArgs(int argc,
                              ->default_value(config_->debug.enable_automatic_ir_metadata)
                              ->implicit_value(true),
                          "Enable automatic IR metadata (debug builds only).");
+  opt_desc.add_options()(
+      "enable-gpu-code-compilation-cache",
+      po::value<bool>(&config_->debug.enable_gpu_code_compilation_cache)
+          ->default_value(config_->debug.enable_gpu_code_compilation_cache)
+          ->implicit_value(true),
+      "Enable GPU compilation code caching.");
 
   // storage
   opt_desc.add_options()(

omniscidb/QueryEngine/NativeCodegen.cpp

@@ -385,7 +385,7 @@ std::shared_ptr<CompilationContext> Executor::optimizeAndCodegenGPU(
 
   auto key = get_code_cache_key(query_func, cgen_state_.get());
   auto cached_code = Executor::gpu_code_accessor->get_value(key);
-  if (cached_code) {
+  if (config_->debug.enable_gpu_code_compilation_cache && cached_code) {
     return cached_code;
   }
 

omniscidb/Shared/Config.h

@@ -171,6 +171,7 @@ struct DebugConfig {
   std::string build_ra_cache = "";
   std::string use_ra_cache = "";
   bool enable_automatic_ir_metadata = true;
+  bool enable_gpu_code_compilation_cache = true;
   std::string log_dir = "hdk_log";
 };
 

python/pyhdk/hdk.py

@@ -1750,7 +1750,7 @@ def schema(self):
         """
         pass
 
-    def run(self):
+    def run(self, **kwargs):
         """
         Run query with the current node as a query root node.
 
@@ -2624,6 +2624,7 @@ def ntile(self, tile_count):
 def init(**kwargs):
     if init._instance is None:
         init._instance = HDK(**kwargs)
+    # TODO: kwargs are ignored here
     return init._instance
 
 

python/tests/conftest.py

@@ -0,0 +1,38 @@
+#
+# Copyright 2022 Intel Corporation.
+#
+# SPDX-License-Identifier: Apache-2.0
+
+
+from pytest import fixture
+from dataclasses import dataclass
+
+
+def pytest_addoption(parser):
+    parser.addoption(
+        "--device", action="store", default="CPU", help="Choose device type (CPU/GPU)."
+    )
+
+
+@dataclass
+class ExecutionConfig:
+    enable_heterogeneous: bool
+    device_type: str
+
+
+cpu_cfg = ExecutionConfig(False, "CPU")
+gpu_cfg = ExecutionConfig(False, "GPU")
+het_cfg = ExecutionConfig(True, "AUTO")
+
+
+def get_execution_config(device_type: str):
+    if device_type.lower() == "cpu":
+        return cpu_cfg
+    if device_type.lower() == "gpu":
+        return gpu_cfg
+    raise ValueError("Unsupported exeuction config: " + str(device_type))
+
+
+@fixture
+def exe_cfg(request):
+    return get_execution_config(request.config.getoption("--device"))