feat: add is_addressable for pjrt devices (#978)

avik-pal · web-flow · commit 35b54863f5f9 · 2025-03-20T18:06:10.000Z
diff --git a/deps/ReactantExtra/API.cpp b/deps/ReactantExtra/API.cpp
@@ -2384,3 +2384,7 @@ extern "C" void ifrt_hlo_module_cost_analysis_properties(
 }
 
 #pragma endregion
+
+extern "C" bool pjrt_device_is_addressable(PjRtDevice *device) {
+  return device->IsAddressable();
+}

Original file line number	Diff line number	Diff line change
`@@ -2384,3 +2384,7 @@ extern "C" void ifrt_hlo_module_cost_analysis_properties(`
`2384`	`2384`	`}`
`2385`	`2385`
`2386`	`2386`	`#pragma endregion`
	`2387`	`+`
	`2388`	`+extern "C" bool pjrt_device_is_addressable(PjRtDevice *device) {`
	`2389`	`+ return device->IsAddressable();`
	`2390`	`+}`