remove unused header content in FFI (#396)

joelberkeley · May 4, 2024 · f168fce · f168fce
1 parent 2d549df
commit f168fce
Show file tree

Hide file tree

Showing 24 changed files with 14 additions and 456 deletions.
diff --git a/backend/VERSION b/backend/VERSION
@@ -1 +1 @@
-0.0.10
+0.0.11
diff --git a/backend/src/ffi.h b/backend/src/ffi.h
diff --git a/backend/src/tensorflow/compiler/xla/client/client_library.cpp b/backend/src/tensorflow/compiler/xla/client/client_library.cpp
@@ -16,8 +16,8 @@ limitations under the License.
 #include "tensorflow/compiler/xla/client/client_library.h"
 #include "tensorflow/compiler/xla/client/local_client.h"
 
+#include "local_client.h"
 #include "../../../stream_executor/platform.h"
-#include "client_library.h"
 
 extern "C" {
     LocalClient* ClientLibrary_GetOrCreateLocalClient(

diff --git a/backend/src/tensorflow/compiler/xla/client/client_library.h b/backend/src/tensorflow/compiler/xla/client/client_library.h
diff --git a/backend/src/tensorflow/compiler/xla/client/global_data.cpp b/backend/src/tensorflow/compiler/xla/client/global_data.cpp
diff --git a/backend/src/tensorflow/compiler/xla/client/lib/arithmetic.h b/backend/src/tensorflow/compiler/xla/client/lib/arithmetic.h
diff --git a/backend/src/tensorflow/compiler/xla/client/lib/constants.h b/backend/src/tensorflow/compiler/xla/client/lib/constants.h
diff --git a/backend/src/tensorflow/compiler/xla/client/lib/math.h b/backend/src/tensorflow/compiler/xla/client/lib/math.h
diff --git a/backend/src/tensorflow/compiler/xla/client/lib/matrix.h b/backend/src/tensorflow/compiler/xla/client/lib/matrix.h
diff --git a/backend/src/tensorflow/compiler/xla/client/lib/prng.cpp b/backend/src/tensorflow/compiler/xla/client/lib/prng.cpp
@@ -17,7 +17,6 @@ limitations under the License.
 
 #include "../../shape.h"
 #include "../xla_builder.h"
-#include "prng.h"
 
 xla::BitGeneratorTy BitGenerator(int bit_generator) {
     xla::BitGeneratorTy bit_generator_;
@@ -37,6 +36,15 @@ xla::BitGeneratorTy BitGenerator(int bit_generator) {
 }
 
 extern "C" {
+    struct RngOutput {
+        XlaOp* value;
+        XlaOp* state;
+    };
+
+    void delete_RngOutput(RngOutput* rngOutput) {
+        free(rngOutput);
+    }
+
     RngOutput* UniformFloatingPointDistribution(
         XlaOp& key,
         XlaOp& initial_state,

diff --git a/backend/src/tensorflow/compiler/xla/client/lib/prng.h b/backend/src/tensorflow/compiler/xla/client/lib/prng.h
diff --git a/backend/src/tensorflow/compiler/xla/client/local_client.cpp b/backend/src/tensorflow/compiler/xla/client/local_client.cpp
@@ -17,7 +17,9 @@ limitations under the License.
 #include "tensorflow/compiler/xla/client/local_client.h"
 
 #include "../literal.h"
+#include "global_data.h"
 #include "local_client.h"
+#include "xla_computation.h"
 
 extern "C" {
     GlobalData* LocalClient_TransferToServer(LocalClient& client, Literal& literal) {

diff --git a/backend/src/tensorflow/compiler/xla/client/local_client.h b/backend/src/tensorflow/compiler/xla/client/local_client.h
@@ -13,19 +13,6 @@ 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.
 */
-#include "../literal.h"
-#include "global_data.h"
-#include "xla_computation.h"
-
 extern "C" {
     struct LocalClient;
-
-    GlobalData* LocalClient_TransferToServer(LocalClient& client, Literal& literal);
-
-    Literal* LocalClient_ExecuteAndTransfer(
-        LocalClient& client,
-        XlaComputation& computation,
-        GlobalData** arguments,
-        int arguments_len
-    );
 }
diff --git a/backend/src/tensorflow/compiler/xla/client/xla_builder.cpp b/backend/src/tensorflow/compiler/xla/client/xla_builder.cpp
@@ -24,6 +24,7 @@ limitations under the License.
 #include "tensorflow/compiler/xla/xla_data.pb.h"
 
 #include "../literal.h"
+#include "../shape.h"
 #include "../xla_data.pb.h"
 #include "xla_builder.h"
 #include "xla_computation.h"