46 lines
1.4 KiB
Diff
46 lines
1.4 KiB
Diff
|
From 0d88ac9b06c8bc78db817d85e90cd60d38e6561a Mon Sep 17 00:00:00 2001
|
||
|
|
From: Hugo Mano <hugo@zml.ai>
|
||
|
|
Date: Mon, 3 Nov 2025 16:54:54 +0100
|
||
|
|
Subject: [PATCH] PjRT C API: make PJRT FFI C extension header compliant
|
||
|
|
|
||
|
|
|
||
|
|
XLA PR: https://github.com/openxla/xla/pull/33470
|
||
|
|
|
||
|
|
---
|
||
|
|
xla/pjrt/c/pjrt_c_api_ffi_extension.h | 8 ++++----
|
||
|
|
1 file changed, 4 insertions(+), 4 deletions(-)
|
||
|
|
|
||
|
|
diff --git a/xla/pjrt/c/pjrt_c_api_ffi_extension.h b/xla/pjrt/c/pjrt_c_api_ffi_extension.h
|
||
|
|
index e756650911..33b78238b9 100644
|
||
|
|
--- a/xla/pjrt/c/pjrt_c_api_ffi_extension.h
|
||
|
|
+++ b/xla/pjrt/c/pjrt_c_api_ffi_extension.h
|
||
|
|
@@ -32,13 +32,13 @@ extern "C" {
|
||
|
|
// See: https://en.wikipedia.org/wiki/Foreign_function_interface
|
||
|
|
#define PJRT_API_FFI_EXTENSION_VERSION 3
|
||
|
|
|
||
|
|
-struct PJRT_FFI_Type_Info {
|
||
|
|
+typedef struct PJRT_FFI_Type_Info {
|
||
|
|
void (*deleter)(void* object);
|
||
|
|
void (*serialize)(); // placeholder for future use
|
||
|
|
void (*deserialize)(); // placeholder for future use
|
||
|
|
-};
|
||
|
|
+} PJRT_FFI_Type_Info;
|
||
|
|
|
||
|
|
-struct PJRT_FFI_Type_Register_Args {
|
||
|
|
+typedef struct PJRT_FFI_Type_Register_Args {
|
||
|
|
size_t struct_size;
|
||
|
|
PJRT_Extension_Base* extension_start;
|
||
|
|
|
||
|
|
@@ -46,7 +46,7 @@ struct PJRT_FFI_Type_Register_Args {
|
||
|
|
size_t type_name_size;
|
||
|
|
int64_t type_id; // in-out
|
||
|
|
PJRT_FFI_Type_Info* type_info;
|
||
|
|
-};
|
||
|
|
+} PJRT_FFI_Type_Register_Args;
|
||
|
|
PJRT_DEFINE_STRUCT_TRAITS(PJRT_FFI_Type_Register_Args, type_info);
|
||
|
|
|
||
|
|
// Registers external type in a static type registry. If `type_id` is set to `0`
|
||
|
|
--
|
||
|
|
2.50.1 (Apple Git-155)
|
||
|
|
|