Skip to content

[SYCL][HIP] Support of AMD matrix core instructions #11485

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

Merged
merged 56 commits into from
Oct 30, 2023

Conversation

mmoadeli
Copy link
Contributor

@mmoadeli mmoadeli commented Oct 10, 2023

  • Support one block AMD matrix core instructions for __gfx90a__ architecture.
  • Supports __builtin_amdgcn_mfma_i32_32x32x8i8, __builtin_amdgcn_mfma_i32_16x16x16i8, __builtin_amdgcn_mfma_f64_16x16x4f64, __builtin_amdgcn_mfma_f32_32x32x8bf16_1k, __builtin_amdgcn_mfma_f32_16x16x16bf16_1k, __builtin_amdgcn_mfma_f32_32x32x8f16 and __builtin_amdgcn_mfma_f32_16x16x16f16 instructions.
  • Add HIP matrix core support into joint_matrix documentation.

Should be merged after

experimental namespace
As part of the effort to move joint matrix from experimental namespace to supported. A review of the API is being done as part of intel#7964. This results in the following changes in the syntax:
1- Add Td to joint_matrix_mad as Tc can be different from Td on the GPU,
Now, we make D as an input argument to mad.
2-  Change “packed” to ext_intel_packed:
3-  Move EWOps (get_wi_data, wi_element, get_coord) to detail namespace)
4- add const to joint_matrix in store and mad
5 - add joint_matrix_copy/assignment function
6- add apply with coordination (change existing tests)
7- change get_coord vector type from int32_t to size_t
8- delete explicitly both = and copy ctor.
@mmoadeli mmoadeli temporarily deployed to WindowsCILock October 10, 2023 10:08 — with GitHub Actions Inactive
@mmoadeli mmoadeli temporarily deployed to WindowsCILock October 24, 2023 08:14 — with GitHub Actions Inactive
Use same code for `copy`, `fill` and `apply`.
Remove `-DSYCL_EXT_ONEAPI_MATRIX_VERSION=4`
@mmoadeli mmoadeli requested a review from JackAKirk October 24, 2023 12:32
@mmoadeli mmoadeli temporarily deployed to WindowsCILock October 24, 2023 12:32 — with GitHub Actions Inactive
@mmoadeli mmoadeli temporarily deployed to WindowsCILock October 24, 2023 12:56 — with GitHub Actions Inactive
@mmoadeli mmoadeli requested a review from JackAKirk October 24, 2023 12:58
@mmoadeli mmoadeli temporarily deployed to WindowsCILock October 24, 2023 13:20 — with GitHub Actions Inactive
@mmoadeli mmoadeli temporarily deployed to WindowsCILock October 24, 2023 15:29 — with GitHub Actions Inactive
@mmoadeli mmoadeli temporarily deployed to WindowsCILock October 24, 2023 15:56 — with GitHub Actions Inactive
Copy link
Contributor

@dkhaldi dkhaldi left a comment

Choose a reason for hiding this comment

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

LGTM

@mmoadeli mmoadeli temporarily deployed to WindowsCILock October 24, 2023 16:32 — with GitHub Actions Inactive
@mmoadeli mmoadeli temporarily deployed to WindowsCILock October 24, 2023 18:12 — with GitHub Actions Inactive
Copy link
Contributor

@gmlueck gmlueck left a comment

Choose a reason for hiding this comment

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

spec changes OK.

@againull againull merged commit 31481ce into intel:sycl Oct 30, 2023
maarquitos14 pushed a commit that referenced this pull request Oct 31, 2023
* Support one block AMD matrix core instructions for `__gfx90a__`
architecture.
* Supports `__builtin_amdgcn_mfma_i32_32x32x8i8`,
`__builtin_amdgcn_mfma_i32_16x16x16i8`,
`__builtin_amdgcn_mfma_f64_16x16x4f64`,
`__builtin_amdgcn_mfma_f32_32x32x8bf16_1k`,
`__builtin_amdgcn_mfma_f32_16x16x16bf16_1k`,
`__builtin_amdgcn_mfma_f32_32x32x8f16` and
`__builtin_amdgcn_mfma_f32_16x16x16f16` instructions.
* Add HIP matrix core support into joint_matrix documentation.

Should be merged after
- #11215

---------

Co-authored-by: Bing1 Yu <[email protected]>
Co-authored-by: mmoadeli <[email protected]>
@mmoadeli mmoadeli deleted the amd-joint-matrix branch November 1, 2023 16:23
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.

9 participants