diff --git a/CMakeLists.txt b/CMakeLists.txt
index 6b7eade8b..346a59486 100644
--- a/CMakeLists.txt
+++ b/CMakeLists.txt
@@ -139,6 +139,7 @@ if (NS_BTLA_UT)
 endif()
 include(FindOpenMP)
 
+set(BTLA_USE_OPENMP ON)
 add_subdirectory(bestla)
 
 add_subdirectory(neural_speed)
diff --git a/bestla/bestla/ut/kernel_wrapper.cpp b/bestla/bestla/ut/kernel_wrapper.cpp
index 76327deb3..26765c06c 100644
--- a/bestla/bestla/ut/kernel_wrapper.cpp
+++ b/bestla/bestla/ut/kernel_wrapper.cpp
@@ -201,14 +201,16 @@ class UT_LayerNormalization {
  public:
   UT_LayerNormalization() {
     UT_START();
-    ut<float, BTLA_ISA::AVX512F>(4096, false, true, true);
-    ut<float, BTLA_ISA::AVX512F>(4096, false, false, false);
-    ut<float, BTLA_ISA::AVX512F>(111, false, true, true);
-    ut<float, BTLA_ISA::AVX512F>(111, true, true, true);
+    CheckISA(AVX2);
     ut<float, BTLA_ISA::AVX2>(4096, false, true, true);
     ut<float, BTLA_ISA::AVX2>(4096, false, false, false);
     ut<float, BTLA_ISA::AVX2>(111, false, true, true);
     ut<float, BTLA_ISA::AVX2>(111, true, true, true);
+    CheckISA(AVX512F);
+    ut<float, BTLA_ISA::AVX512F>(4096, false, true, true);
+    ut<float, BTLA_ISA::AVX512F>(4096, false, false, false);
+    ut<float, BTLA_ISA::AVX512F>(111, false, true, true);
+    ut<float, BTLA_ISA::AVX512F>(111, true, true, true);
   }
   template <typename T, BTLA_ISA ISA>
   void ut(int norm_size, bool simplified, bool hasscale, bool hasbias) {