iree-org / iree

A retargetable MLIR-based machine learning compiler and runtime toolkit.
http://iree.dev/
Apache License 2.0
2.85k stars 620 forks source link

Add build and test workflow for PJRT plugin in pkgci #19222

Open PragmaTwice opened 2 days ago

PragmaTwice commented 2 days ago

This changes are for adding the build workflow for iree-pjrt-plugin-* packages in build_tools/pkgci scripts.

Also, it enables the build of package iree-pjrt-plugin-cpu in Github Actions (pkgci.yaml).

Some other changes are for fixing some build problems:

It closes #19221.

ci-exactly: build_packages, test_pjrt

PragmaTwice commented 1 day ago

Hi @ScottTodd , thanks for your kindly review. Learn a lot.

I've refactored it to a separate PJRT build & test workflow now, feel free to give suggestions : )

Currently only the CPU plugin is enabled in CI (but the workflow is extensible itself, can just uncomment some matrix item) since:

ScottTodd commented 13 hours ago

I can also send you an invite to iree-org so the checks will run automatically for your PRs if you want.

PragmaTwice commented 13 hours ago

Thank you for your quick actions!

I can also send you an invite to iree-org so the checks will run automatically for your PRs if you want.

Ahh thank you. That would be great. I'll accept the invitation : )

ScottTodd commented 13 hours ago

No problem. Caught me just before making dinner 😀 . Invite sent. Docs for access levels are at https://iree.dev/developers/general/contributing/#obtaining-commit-access