From fcf50ca0810b3d68a8db96ace8d41382547173c5 Mon Sep 17 00:00:00 2001
From: shahasad <43590019+shahasad@users.noreply.github.com>
Date: Mon, 21 Oct 2019 08:46:28 -0700
Subject: [PATCH] Fix nuget mklml pipeline (#2204)

* some fixes on nuget CPU pipeline

* revert d738c89536a06b09a61532bb887f2bcc2182dafb

* fix for MKLML package

* fix if else
---
 .../templates/get-nuget-package-version-as-variable.yml  | 9 ++++++++-
 1 file changed, 8 insertions(+), 1 deletion(-)

diff --git a/tools/ci_build/github/azure-pipelines/nuget/templates/get-nuget-package-version-as-variable.yml b/tools/ci_build/github/azure-pipelines/nuget/templates/get-nuget-package-version-as-variable.yml
index f31a4ad1b3..58d2a232ca 100644
--- a/tools/ci_build/github/azure-pipelines/nuget/templates/get-nuget-package-version-as-variable.yml
+++ b/tools/ci_build/github/azure-pipelines/nuget/templates/get-nuget-package-version-as-variable.yml
@@ -14,7 +14,11 @@ steps:
         set ortversion=!filename:~25!
         IF "!filename:~25,3!"=="Gpu" (
           set ortversion=!filename:~29!
-        )  
+        ) ELSE (
+          IF "!filename:~25,5!"=="MKLML" (
+            set ortversion=!filename:~31!
+          )
+        ) 
 
         @echo ortversion is !ortversion!
         @echo ##vso[task.setvariable variable=NuGetPackageVersionNumber;]!ortversion!
@@ -32,5 +36,8 @@ steps:
       if [ "${filename:25:3}" = "Gpu" ]
       then
          ortversion=${filename:29}
+      elif [ "${filename:25:5}" = "MKLML" ]
+      then
+         ortversion=${filename:31}
       fi
       echo "##vso[task.setvariable variable=NuGetPackageVersionNumber;]$ortversion"