From 5ddf3a215b21c48e2f8641275655b046402dc13d Mon Sep 17 00:00:00 2001
From: Ryan Lai <rylai@microsoft.com>
Date: Thu, 29 Aug 2019 15:51:41 -0700
Subject: [PATCH] Can't use DXGI to print adapter name of VPU (#274)

* Fix vpu break, can't print adaptername using dxcore

* Spacing
---
 .../src/LearningModelDeviceHelper.cpp         | 21 ++++++++++++-------
 Tools/WinMLRunner/src/OutputHelper.h          |  6 +++---
 Tools/WinMLRunner/src/Run.cpp                 |  1 -
 3 files changed, 17 insertions(+), 11 deletions(-)

diff --git a/Tools/WinMLRunner/src/LearningModelDeviceHelper.cpp b/Tools/WinMLRunner/src/LearningModelDeviceHelper.cpp
index 4fdc2324..fdd2b707 100644
--- a/Tools/WinMLRunner/src/LearningModelDeviceHelper.cpp
+++ b/Tools/WinMLRunner/src/LearningModelDeviceHelper.cpp
@@ -6,6 +6,7 @@
 #include <Windows.Graphics.DirectX.Direct3D11.interop.h>
 #include "Windows.AI.MachineLearning.Native.h"
 #include <codecvt>
+#include "OutputHelper.h"
 using namespace winrt::Windows::Graphics::DirectX::Direct3D11;
 
 #ifdef DXCORE_SUPPORTED_BUILD
@@ -83,11 +84,14 @@ void PopulateLearningModelDeviceList(CommandLineArgs& args, std::vector<Learning
                     com_ptr<IInspectable> inspectableDevice;
                     hr = CreateDirect3D11DeviceFromDXGIDevice(dxgiDevice.get(), inspectableDevice.put());
                     THROW_IF_FAILED(hr);
-                    deviceList.push_back({
+                    LearningModelDeviceWithMetadata learningModelDeviceWithMetadata =
+                        {
                         LearningModelDevice::CreateFromDirect3D11Device(inspectableDevice.as<IDirect3DDevice>()),
                         deviceType,
                         deviceCreationLocation
-                        });
+                        };
+                    OutputHelper::PrintLearningModelDevice(learningModelDeviceWithMetadata);
+                    deviceList.push_back(learningModelDeviceWithMetadata);
                 }
 #ifdef DXCORE_SUPPORTED_BUILD
                 else if ((TypeHelper::GetWinmlDeviceKind(deviceType) != LearningModelDeviceKind::Cpu) &&
@@ -216,11 +220,14 @@ void PopulateLearningModelDeviceList(CommandLineArgs& args, std::vector<Learning
 #endif
                 else
                 {
-                    deviceList.push_back({
-                        LearningModelDevice(TypeHelper::GetWinmlDeviceKind(deviceType)),
-                        deviceType,
-                        deviceCreationLocation
-                        });
+                    LearningModelDeviceWithMetadata learningModelDeviceWithMetadata = 
+                    { 
+                        LearningModelDevice( TypeHelper::GetWinmlDeviceKind(deviceType)),
+                                             deviceType, 
+                                             deviceCreationLocation 
+                    };
+                    OutputHelper::PrintLearningModelDevice(learningModelDeviceWithMetadata);
+                    deviceList.push_back(learningModelDeviceWithMetadata);
                 }
             }
             catch (...)
diff --git a/Tools/WinMLRunner/src/OutputHelper.h b/Tools/WinMLRunner/src/OutputHelper.h
index b5f575a1..13b666de 100644
--- a/Tools/WinMLRunner/src/OutputHelper.h
+++ b/Tools/WinMLRunner/src/OutputHelper.h
@@ -165,11 +165,11 @@ class OutputHelper
         std::cout << std::endl;
     }
 
-    void PrintLearningModelDevice(const LearningModelDeviceWithMetadata& device)
+    static void PrintLearningModelDevice(const LearningModelDeviceWithMetadata& device)
     {
         if (device.DeviceType == DeviceType::CPU)
         {
-            std::cout << "\nCreating Session with CPU device" << std::endl;
+            std::cout << "\nCreated LearningModelDevice with CPU device" << std::endl;
             return;
         }
 
@@ -185,7 +185,7 @@ class OutputHelper
             DXGI_ADAPTER_DESC description;
             if (SUCCEEDED(adapter->GetDesc(&description)))
             {
-                std::wcout << L"\nCreating Session with GPU: " << description.Description << std::endl;
+                std::wcout << L"\nCreated LearningModelDevice with GPU: " << description.Description << std::endl;
             }
         }
         else
diff --git a/Tools/WinMLRunner/src/Run.cpp b/Tools/WinMLRunner/src/Run.cpp
index e42f8aee..646411fd 100644
--- a/Tools/WinMLRunner/src/Run.cpp
+++ b/Tools/WinMLRunner/src/Run.cpp
@@ -180,7 +180,6 @@ HRESULT CreateSession(LearningModelSession& session, LearningModel& model, const
     }
     try
     {
-        output.PrintLearningModelDevice(learningModelDevice);
         CreateSessionConsideringSupportForSessionOptions(session, model, profiler, args, learningModelDevice);
     }
     catch (hresult_error hr)