Added new BO strategies to interface

KernelTuner · Oct 25, 2024 · 6281a0c · 6281a0c
1 parent 208fe7b
commit 6281a0c
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 1 deletion.
diff --git a/kernel_tuner/interface.py b/kernel_tuner/interface.py
@@ -50,6 +50,10 @@
 from kernel_tuner.strategies import (
     basinhopping,
     bayes_opt,
+    bayes_opt_alt_BOTorch,
+    bayes_opt_GPyTorch,
+    bayes_opt_GPyTorch_lean,
+    bayes_opt_old,
     brute_force,
     diff_evo,
     dual_annealing,

diff --git a/test/context.py b/test/context.py
@@ -47,7 +47,8 @@
     cupy_present = False
 
 try:
-
+    import cuda
+    print(cuda)
     cuda_present = True
 except Exception:
     cuda_present = False