Require MKL for LSH (with rotation)

This change was requested to avoid silently falling back to slower BLAS implementations.
marian-nmt · Sep 13, 2022 · d713aec · d713aec
1 parent 70372fd
commit d713aec
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/layers/lsh.cpp b/src/layers/lsh.cpp
@@ -4,7 +4,7 @@
 
 #include "3rd_party/faiss/utils/hamming.h"
 
-#if BLAS_FOUND
+#if defined(BLAS_FOUND) && defined(MKL_FOUND)
 #include "3rd_party/faiss/VectorTransform.h"
 #endif
 
@@ -20,7 +20,7 @@ int bytesPerVector(int nBits) {
 }
 
 void fillRandomRotationMatrix(Tensor output, Ptr<Allocator> allocator) {
-#if BLAS_FOUND
+#if defined(BLAS_FOUND) && defined(MKL_FOUND)
   int nRows = output->shape()[-2];
   int nBits = output->shape()[-1];