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