[BUILD] Bug fix in the TensorFlow native packaging plugin.

eaplatanios · eaplatanios · commit 6ec8fbdd82b7 · 2018-06-17T10:25:26.000-07:00
diff --git a/project/TensorFlowNativePackage.scala b/project/TensorFlowNativePackage.scala
@@ -113,7 +113,7 @@ object TensorFlowNativePackage extends AutoPlugin {
     case (LINUX_GPU_x86_64, "nightly") => s"$tfLibNightlyUrlPrefix/TYPE=gpu-linux/lastSuccessfulBuild/artifact/lib_package/${tfLibFilename(platform)}"
     case (DARWIN_x86_64, "nightly") => s"$tfLibNightlyUrlPrefix/TYPE=mac-slave/lastSuccessfulBuild/artifact/lib_package/${tfLibFilename(platform)}"
     case (LINUX_x86_64, v) => s"$tfLibUrlPrefix/libtensorflow-cpu-${platform.name}-$v.tar.gz"
-    case (LINUX_GPU_x86_64, v) => s"$tfLibUrlPrefix/libtensorflow-gpu-${platform.name}-$v.tar.gz"
+    case (LINUX_GPU_x86_64, v) => s"$tfLibUrlPrefix/libtensorflow-gpu-${LINUX_x86_64.name}-$v.tar.gz"
     case (DARWIN_x86_64, v) => s"$tfLibUrlPrefix/libtensorflow-cpu-${platform.name}-$v.tar.gz"
   }
 

Original file line number	Diff line number	Diff line change
`@@ -113,7 +113,7 @@ object TensorFlowNativePackage extends AutoPlugin {`
`113`	`113`	`case (LINUX_GPU_x86_64, "nightly") => s"$tfLibNightlyUrlPrefix/TYPE=gpu-linux/lastSuccessfulBuild/artifact/lib_package/${tfLibFilename(platform)}"`
`114`	`114`	`case (DARWIN_x86_64, "nightly") => s"$tfLibNightlyUrlPrefix/TYPE=mac-slave/lastSuccessfulBuild/artifact/lib_package/${tfLibFilename(platform)}"`
`115`	`115`	`case (LINUX_x86_64, v) => s"$tfLibUrlPrefix/libtensorflow-cpu-${platform.name}-$v.tar.gz"`
`116`		`- case (LINUX_GPU_x86_64, v) => s"$tfLibUrlPrefix/libtensorflow-gpu-${platform.name}-$v.tar.gz"`
	`116`	`+ case (LINUX_GPU_x86_64, v) => s"$tfLibUrlPrefix/libtensorflow-gpu-${LINUX_x86_64.name}-$v.tar.gz"`
`117`	`117`	`case (DARWIN_x86_64, v) => s"$tfLibUrlPrefix/libtensorflow-cpu-${platform.name}-$v.tar.gz"`
`118`	`118`	`}`
`119`	`119`