From 6281a0c012ff8d71da2a6dd2c032db9722805518 Mon Sep 17 00:00:00 2001 From: Floris-Jan Willemsen Date: Fri, 25 Oct 2024 16:18:30 -0700 Subject: [PATCH] Added new BO strategies to interface --- kernel_tuner/interface.py | 4 ++++ test/context.py | 3 ++- 2 files changed, 6 insertions(+), 1 deletion(-) diff --git a/kernel_tuner/interface.py b/kernel_tuner/interface.py index 06b5058f..a557ae58 100644 --- 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 index e9959176..e7bb7cbf 100644 --- 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