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

general transpose for oneAPI #1165

Open
wants to merge 1 commit into
base: main
Choose a base branch
from
Open

Conversation

jmitrevs
Copy link
Contributor

Description

This uses the model of PR #1124 to add a general transpose for oneAPI. It doesn't check the synthesis performance. If needed, enable ifs can be added to use the old implementation, but I have avoided that complication unless it proves to be necessary.

Type of change

  • New feature (non-breaking change which adds functionality)

Tests

Just added oneAPI to the test from #1124

Checklist

  • I have read the guidelines for contributing.
  • I have commented my code, particularly in hard-to-understand areas.
  • I have made corresponding changes to the documentation.
  • My changes generate no new warnings.
  • I have installed and run pre-commit on the files I edited or added.
  • I have added tests that prove my fix is effective or that my feature works.

transpose_include_list = ['nnet_utils/nnet_transpose.h', 'nnet_utils/nnet_transpose_stream.h']


def permute_config_gen(name: str, shape: tuple[int, ...], perm: tuple[int, ...]):
Copy link
Contributor

Choose a reason for hiding this comment

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

This is identical to the one introduced by the #1124, it should be placed in some utility module to be shared by all backends.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

2 participants