Skip to content

Add structural sparsity gemm instruction tests#1309

Merged
aosewski merged 7 commits intodevelopfrom
jakpiase/smfmac_test
Jun 27, 2024
Merged

Add structural sparsity gemm instruction tests#1309
aosewski merged 7 commits intodevelopfrom
jakpiase/smfmac_test

Conversation

@jakpiase
Copy link
Contributor

Added single wavefront tests of AMD's structural sparsity matrix multiplication instructions. Currently following instructions are supported:

v_smfmac_f32_16x16x32f16
v_smfmac_f32_16x16x32bf16
v_smfmac_f32_32x32x16f16
v_smfmac_f32_32x32x16bf16

Copy link
Collaborator

@aosewski aosewski left a comment

Choose a reason for hiding this comment

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

In general looks good. But please write unit tests using gtest.

Copy link
Collaborator

@aosewski aosewski left a comment

Choose a reason for hiding this comment

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

Just a few minor changes to tests.

@aosewski aosewski merged commit ed21948 into develop Jun 27, 2024
@jakpiase jakpiase deleted the jakpiase/smfmac_test branch June 27, 2024 09:31
if(GPU_TARGETS MATCHES "gfx11")
add_subdirectory(wmma_op)
endif()
if(GPU_TARGETS MATCHES "gfx942")
Copy link
Contributor

Choose a reason for hiding this comment

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

I guess this is problematic because if one also targets other platforms, this will also get built and causing unsupported instructions error

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.

4 participants