Got things to compile on Windows.

Author: eaplatanios

.gitignore

@@ -21,6 +21,7 @@ project/plugins/project/
 
 # Bloop Specific
 .bloop
+.bsp
 
 # Metals Specific
 .metals

README.md

@@ -207,6 +207,8 @@ command:
 sbt jni/cross:nativeCrossCompile
 ```
 
+CUDA Compute Capabilities: 3.5,7.0,7.5,8.0,8.6
+
 Compile the TensorFlow dynamic libraries from source using:
 
 ```bash

build.sbt

@@ -170,7 +170,7 @@ lazy val jni = (project in file("./modules/jni"))
       sourceDirectory in nativeCompile := sourceDirectory.value / "main" / "native",
       target in nativeCompile := target.value / "native" / nativePlatform.value,
       target in JniCross := target.value / "native",
-      nativePlatforms in JniCross := Set(LINUX, WINDOWS_CPU, WINDOWS_GPU, DARWIN),
+      nativePlatforms in JniCross := Set(LINUX, WINDOWS, WINDOWS_CPU, DARWIN),
       tfBinaryVersion in JniCross := tensorFlowVersion,
       // Specify the order in which the different compilation tasks are executed.
       nativeCompile := nativeCompile.dependsOn(generateTensorOps).value,

modules/jni/src/main/native/ops/beam_search_ops.cc

@@ -28,6 +28,11 @@ limitations under the License.
 #include "tensorflow/core/framework/shape_inference.h"
 #include "tensorflow/core/util/work_sharder.h"
 
+// Hack to fix compilation on Windows.
+#define EXPAND(x) x
+#define TF_NEW_ID_FOR_INIT_2(m, c, ...) EXPAND(m(c, __VA_ARGS__)) // L145 selective_registration.h
+#define TF_EXTRACT_KERNEL_NAME_IMPL(m, ...) EXPAND(m(__VA_ARGS__)) // L1431 op_kernel.h
+
 namespace tensorflow {
 
 using shape_inference::DimensionHandle;

modules/jni/src/main/scala/org/platanios/tensorflow/jni/TensorFlow.scala

@@ -56,7 +56,7 @@ object TensorFlow {
     val name = System.getProperty("os.name").toLowerCase
     if (name.contains("linux")) "linux"
     else if (name.contains("os x") || name.contains("darwin")) "darwin"
-    else if (name.contains("windows")) "windows-cpu"
+    else if (name.contains("windows")) "windows"
     else name.replaceAll("\\s", "")
   }
 

project/JniCrossPackage.scala

@@ -59,7 +59,7 @@ object JniCrossPackage extends AutoPlugin {
   import JniNative.autoImport._
 
   lazy val settings: Seq[Setting[_]] = Seq(
-    nativePlatforms := Set(LINUX, WINDOWS_CPU, WINDOWS_GPU, DARWIN),
+    nativePlatforms := Set(LINUX, WINDOWS, WINDOWS_CPU, DARWIN),
     target := (target in Compile).value / "native",
     nativeLibPath := {
       val targetDir = (target in nativeCrossCompile).value
@@ -106,7 +106,7 @@ object JniCrossPackage extends AutoPlugin {
               // TODO: Figure out the right cross-compilation story.
               // For Windows, we expect the binaries to have already been built and placed in the `bin` and the `lib`
               // subdirectories, because we currently have no way to cross-compile.
-              case WINDOWS_CPU | WINDOWS_GPU =>
+              case WINDOWS | WINDOWS_CPU =>
                 if (!(platformTargetDir / "bin").exists()) {
                   throw new IllegalStateException("The Windows binaries must have already been prebuilt.")
                 }
@@ -255,15 +255,15 @@ object JniCrossPackage extends AutoPlugin {
 
   object LINUX extends Platform {
     override val name       : String = "linux"
-    override val dockerImage: String = "eaplatanios/tensorflow_scala:linux-cpu-x86_64-0.5.3"
+    override val dockerImage: String = "eaplatanios/tensorflow_scala:linux-cpu-x86_64-0.6.0"
   }
 
-  object WINDOWS_CPU extends Platform {
-    override val name: String = "windows-cpu"
+  object WINDOWS extends Platform {
+    override val name: String = "windows"
   }
 
-  object WINDOWS_GPU extends Platform {
-    override val name: String = "windows-gpu"
+  object WINDOWS_CPU extends Platform {
+    override val name: String = "windows-cpu"
   }
 
   object DARWIN extends Platform {

project/TensorFlowNativePackage.scala

@@ -67,13 +67,13 @@ object TensorFlowNativePackage extends AutoPlugin {
 
   def tfLibFilename(platform: Platform): String = platform match {
     case LINUX | DARWIN => "libtensorflow.tar.gz"
-    case WINDOWS_CPU | WINDOWS_GPU => "libtensorflow.zip"
+    case WINDOWS | WINDOWS_CPU => "libtensorflow.zip"
   }
 
   def tfLibUrl(platform: Platform, version: String): String = (platform, version) match {
     case (LINUX, v) => s"$tfLibUrlPrefix-gpu-linux-x86_64-$v.tar.gz"
+    case (WINDOWS, v) => s"$tfLibUrlPrefix-gpu-windows-x86_64-$v.zip"
     case (WINDOWS_CPU, v) => s"$tfLibUrlPrefix-cpu-windows-x86_64-$v.zip"
-    case (WINDOWS_GPU, v) => s"$tfLibUrlPrefix-gpu-windows-x86_64-$v.zip"
     case (DARWIN, v) => s"$tfLibUrlPrefix-cpu-darwin-x86_64-$v.tar.gz"
   }