Merge pull request #1176 from compnerd/x10-spm

build: enable building X10 support with SPM
tensorflow · Jan 6, 2021 · eebf3c7 · eebf3c7
2 parents d2e3b0c + a215cc6
commit eebf3c7
Show file tree

Hide file tree

Showing 14 changed files with 34 additions and 35 deletions.
diff --git a/Package.swift b/Package.swift
@@ -43,13 +43,21 @@ let package = Package(
     .target(
       name: "CTensorFlow",
       dependencies: []),
+    .target(
+      name: "CX10Modules",
+      dependencies: []),
     .target(
       name: "TensorFlow",
       dependencies: [
         "Tensor",
         "PythonKit",
         "CTensorFlow",
+        "CX10Modules",
         .product(name: "Numerics", package: "swift-numerics"),
+      ],
+      swiftSettings: [
+        .define("USING_X10_BACKEND"),
+        .define("DEFAULT_BACKEND_EAGER"),
       ]),
     .target(
       name: "Experimental",

diff --git a/Sources/CX10Modules/include/device_wrapper.h b/Sources/CX10Modules/include/device_wrapper.h
@@ -0,0 +1 @@
+../../CX10/device_wrapper.h
diff --git a/Sources/CX10Modules/include/module.modulemap b/Sources/CX10Modules/include/module.modulemap
@@ -0,0 +1 @@
+../../CX10/module.modulemap
diff --git a/Sources/CX10Modules/include/xla_tensor_tf_ops.h b/Sources/CX10Modules/include/xla_tensor_tf_ops.h
@@ -0,0 +1 @@
+../../CX10/xla_tensor_tf_ops.h
diff --git a/Sources/CX10Modules/include/xla_tensor_wrapper.h b/Sources/CX10Modules/include/xla_tensor_wrapper.h
@@ -0,0 +1 @@
+../../CX10/xla_tensor_wrapper.h
diff --git a/Sources/CX10Modules/shim.c b/Sources/CX10Modules/shim.c
@@ -0,0 +1,15 @@
+// Copyright 2019 The TensorFlow Authors. All Rights Reserved.
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+//     http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+
+extern int _;
diff --git a/Sources/TensorFlow/Core/Device.swift b/Sources/TensorFlow/Core/Device.swift
diff --git a/Sources/TensorFlow/X10/APIs/CrossReplicaSum.swift b/Sources/TensorFlow/X10/APIs/CrossReplicaSum.swift
@@ -0,0 +1 @@
+../../../x10/swift_bindings/apis/CrossReplicaSum.swift
diff --git a/Sources/TensorFlow/X10/APIs/DeviceScope.swift b/Sources/TensorFlow/X10/APIs/DeviceScope.swift
@@ -0,0 +1 @@
+../../../x10/swift_bindings/apis/DeviceScope.swift
diff --git a/Sources/TensorFlow/X10/APIs/RawOpsManual.swift b/Sources/TensorFlow/X10/APIs/RawOpsManual.swift
@@ -0,0 +1 @@
+../../../x10/swift_bindings/apis/RawOpsManual.swift
diff --git a/Sources/TensorFlow/X10/Device.swift b/Sources/TensorFlow/X10/Device.swift
@@ -0,0 +1 @@
+../../x10/swift_bindings/Device.swift
diff --git a/Sources/TensorFlow/X10/RawOpsXLAGenerated.swift b/Sources/TensorFlow/X10/RawOpsXLAGenerated.swift
@@ -0,0 +1 @@
+../../x10/swift_bindings/RawOpsXLAGenerated.swift
diff --git a/Sources/TensorFlow/X10/XLAScalarType.swift b/Sources/TensorFlow/X10/XLAScalarType.swift
@@ -0,0 +1 @@
+../../x10/swift_bindings/XLAScalarType.swift
diff --git a/Sources/TensorFlow/X10/XLATensor.swift b/Sources/TensorFlow/X10/XLATensor.swift
@@ -0,0 +1 @@
+../../x10/swift_bindings/XLATensor.swift