Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Add preliminary support for the xcore target. #585

Merged
merged 1 commit into from
Jan 28, 2021
Merged

Conversation

AdamHillier
Copy link
Contributor

@AdamHillier AdamHillier commented Jan 22, 2021

What do these changes do?

This PR adds support for a 'target' flag to the LCE converter, with two values: "arm" (the default), and "xcore" (the XMOS xcore.ai board). For the time being, the only difference between the two is that the "arm" target supports 'same zero' padding, and the binary maxpool op, but the "xcore" target does not.

How Has This Been Tested?

I added a small test case to the Python converter test, but it's not possible to use the end2end tests with the xcore target because the converted models are not compatible with the LCE interpreter (and indeed are not meant to be). Need to work out what to do here.

Benchmark Results

N/A.

Related issue number

N/A.

@AdamHillier
Copy link
Contributor Author

Ideally we'd be able to have separate MLIR tests for the different targets, but I'm not sure our setup currently supports doing that?

@AdamHillier AdamHillier force-pushed the xcore-target branch 3 times, most recently from 922bb6c to 5b5423e Compare January 26, 2021 10:17
@lgeiger lgeiger added this to the 0.5 milestone Jan 27, 2021
@AdamHillier AdamHillier marked this pull request as ready for review January 28, 2021 00:17
@AdamHillier AdamHillier requested a review from a team January 28, 2021 00:17
@lgeiger lgeiger self-requested a review January 28, 2021 09:09
Copy link
Member

@lgeiger lgeiger left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

This looks great!

I only have one small suggestion, but I am also happy to merge this PR as is.

larq_compute_engine/mlir/transforms/prepare_tf.cc Outdated Show resolved Hide resolved
@lgeiger lgeiger added the feature New feature or request label Jan 28, 2021
@AdamHillier AdamHillier enabled auto-merge (squash) January 28, 2021 15:53
@AdamHillier AdamHillier merged commit 2f85ab7 into master Jan 28, 2021
@AdamHillier AdamHillier deleted the xcore-target branch January 28, 2021 19:04
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
feature New feature or request
Projects
None yet
Development

Successfully merging this pull request may close these issues.

3 participants