Filter CPU case for IsFloat16Supported (#2802)

microsoft · Jan 13, 2020 · 6ad3e74 · 6ad3e74
1 parent 8d017f5
commit 6ad3e74
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/winml/lib/Api.Image/DeviceHelpers.cpp b/winml/lib/Api.Image/DeviceHelpers.cpp
@@ -9,6 +9,7 @@
 #include <d3d11on12.h>
 #include <wil/winrt.h>
 #include "inc/DeviceHelpers.h"
+#include "LearningModelDevice.h"
 
 namespace DeviceHelpers {
 constexpr uint32_t c_intelVendorId = 0x8086;
@@ -133,6 +134,10 @@ static HRESULT IsFloat16Blocked(ID3D12Device& device, bool* isBlocked) {
 }
 
 bool IsFloat16Supported(const winrt::Windows::AI::MachineLearning::LearningModelDevice& device) {
+  auto modelImpl = device.as<winmlp::LearningModelDevice>();
+  if (modelImpl->IsCpuDevice()) {
+    return true;
+  }
   winrt::com_ptr<ID3D12Device> d3d12Device;
   if (FAILED(GetD3D12Device(device, d3d12Device.put()))) {
     return false;