Commit51d1341

authored and

committed

Explicitly include GPU FFI support in mosaic custom call definition

Forward compatible header re-export will be removed in next XLA change.PiperOrigin-RevId: 837568729

1 parenta418956 commit51d1341Copy full SHA for 51d1341

File tree

-0

lines changed

-0

lines changed

Lines changed: 1 addition & 0 deletions

Original file line number	Diff line number	Diff line change
`@@ -327,6 +327,7 @@ cc_library(`
`327`	`327`	`"@tsl//tsl/platform:path",`
`328`	`328`	`"@tsl//tsl/profiler/lib:traceme",`
`329`	`329`	`"@xla//xla:executable_run_options",`
	`330`	`+"@xla//xla/backends/gpu:ffi",`
`330`	`331`	`"@xla//xla/ffi",`
`331`	`332`	`"@xla//xla/ffi:ffi_api",`
`332`	`333`	`"@xla//xla/service/gpu/llvm_gpu_backend:nvptx_libdevice_path",`

Lines changed: 1 addition & 0 deletions

Original file line number	Diff line number	Diff line change
`@@ -109,6 +109,7 @@ limitations under the License.`
`109`	`109`	`#include"jaxlib/mosaic/gpu/passes.h"`
`110`	`110`	`#include"jaxlib/mosaic/gpu/serde.h"`
`111`	`111`	`#include"jaxlib/mosaic/gpu/target.h"`
	`112`	`+#include"xla/backends/gpu/ffi.h"`
`112`	`113`	`#include"xla/executable_run_options.h"`
`113`	`114`	`#include"xla/ffi/ffi.h"`
`114`	`115`	`#include"xla/ffi/ffi_api.h"`

Comments

(0)