OSSForks/tensorflow
1
0
Fork 0
mirror of https://github.com/zebrajr/tensorflow.git synced 2025-12-06 12:20:11 +01:00
Code Issues Actions 5 Packages Projects Releases Wiki Activity

[xla:ffi] Add xla::ffi::MakeTypeInfo

Browse Source 
PiperOrigin-RevId: 825230372
This commit is contained in:
Eugene Zhulenev 2025-10-28 15:28:18 -07:00 committed by TensorFlower Gardener
parent 1d829c0f02
commit 3ea8166731
1 changed files with 6 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
third_party/xla/xla/ffi/api/ffi.h vendored
View File

@ -1503,11 +1503,17 @@ inline ThreadPool::ThreadPool(const XLA_FFI_Api* api,
// Type Registration
//===----------------------------------------------------------------------===//
// TODO(ezhulenev): Remove this once everyone migrates to MakeTypeInfo.
template <typename T>
constexpr XLA_FFI_TypeInfo TypeInfo() {
return XLA_FFI_TypeInfo{[](void* ptr) { delete static_cast<T*>(ptr); }};
}
template <typename T>
constexpr XLA_FFI_TypeInfo MakeTypeInfo() {
return XLA_FFI_TypeInfo{[](void* ptr) { delete static_cast<T*>(ptr); }};
}
#define XLA_FFI_REGISTER_TYPE(API, NAME, TYPE_ID, TYPE_INFO) \
XLA_FFI_REGISTER_TYPE_(API, NAME, TYPE_ID, TYPE_INFO, __COUNTER__)
#define XLA_FFI_REGISTER_TYPE_(API, NAME, TYPE_ID, TYPE_INFO, N) \