diff --git a/src/EmbeddingSpMDM.cc b/src/EmbeddingSpMDM.cc index 26c5fc45f..5e4e11b75 100644 --- a/src/EmbeddingSpMDM.cc +++ b/src/EmbeddingSpMDM.cc @@ -1032,11 +1032,11 @@ typename EmbeddingSpMDMKernelSignature:: if (!cpuinfo_initialize()) { throw std::runtime_error("Failed to initialize cpuinfo!"); } -#if defined(__APPLE__) +#if defined(__APPLE__) || defined(_WIN32) if (std::is_same::value && is_bf16_in && std::is_same::value) { throw std::runtime_error( - "Bfloat16 input with float32 output is not yet supported on Apple"); + "Bfloat16 input with float32 output is not yet supported on Apple or Windows"); } #endif if (output_stride == -1) { diff --git a/test/EmbeddingSpMDMTest.cc b/test/EmbeddingSpMDMTest.cc index d4dbac6b3..68962cff2 100644 --- a/test/EmbeddingSpMDMTest.cc +++ b/test/EmbeddingSpMDMTest.cc @@ -145,7 +145,7 @@ TEST_P(EmbeddingSpMDMTest, basicTest) { return; } -#if defined(__APPLE__) +#if defined(__APPLE__) || defined(_WIN32) if (in_type == BFLOAT16 && out_type == FLOAT) { return; }