xla icon indicating copy to clipboard operation
xla copied to clipboard

Published 20 hours ago • verified publisher icon openxla

[PJRT C API] Add some tests for PJRT C API implementation.

Open copybara-service[bot] opened this issue 2 years ago • 0 comments

[PJRT C API] Add some tests for PJRT C API implementation.

  • pjrt_c_api_test contains a basic set of tests.
  • A plugin can run the tests in pjrt_c_api_test via RegisterTestCApiFactory. Added a GPU PJRT plugin pjrt_c_api_gpu.h/cc and its test pjrt_c_api_test_gpu.cc as an example.

copybara-service[bot] avatar Feb 23 '23 19:02 copybara-service[bot]