Skip to content

Support <F8,B,S>, <F8B8,B,S> with BiasB for type TN - #522

Merged
KKyang merged 4 commits into
ROCm:developfrom
solaslin:add-new-fp8-type-api
Jan 4, 2024
Merged

KKyang merged 4 commits into
ROCm:developfrom
solaslin:add-new-fp8-type-api

Conversation

@solaslin

@solaslin solaslin commented Dec 28, 2023 •

Copy link
Copy Markdown
Contributor

This PR resolves basic requirement SWDEV-436059

A_type=fp8 B_type=fp8 D_type=bf16 transa=T transb=N
A_type=fp8 B_type=bf8 D_type=bf16 transa=T transb=N
A_type=fp8 B_type=fp8 D_type=bf16 bias_type=bf16 transa=T transb=N

  • hipBLASLt runtime support
  • Added logic yamls to gfx942, GridBased
  • Fixed a few typos in existing comments
  • Added tensile test to Test common
  • Added gtest

@solaslin solaslin self-assigned this Dec 28, 2023
@solaslin
solaslin force-pushed the add-new-fp8-type-api branch from 08f6292 to 2beae93 Compare December 28, 2023 02:57
@solaslin
solaslin force-pushed the add-new-fp8-type-api branch from 763c72a to e7f30ec Compare December 28, 2023 06:23
Comment on lines 93 to 99

Real precisions 1 bytes dst bf16: &real_precisions_1b_dst_bf16
- *f8_precision_dst_bf16
- *f8bf8_precision_dst_bf16
# - *bf8f8_precision_dst_bf16 # TODO

Real precisions 1 bytes dst fp8: &real_precisions_1b_dst_f8

Copy link
Copy Markdown
Contributor Author

Choose a reason for hiding this comment

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

the third one is not included in this pr but we are gonna support it later. so this is just a placeholder

Comment thread CMakeLists.txt
Comment on lines 131 to 133
if (CMAKE_BUILD_TYPE MATCHES "Debug" AND NOT WIN32)
set (CMAKE_CXX_FLAGS_DEBUG "-O1 ${CMAKE_CXX_FLAGS_DEBUG} -gsplit-dwarf -ggdb" )
set (CMAKE_CXX_FLAGS_DEBUG "-O0 ${CMAKE_CXX_FLAGS_DEBUG} -gsplit-dwarf -ggdb" )
endif()

Copy link
Copy Markdown
Contributor Author

Choose a reason for hiding this comment

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

-O0 can get more detailed info when debugging

Comment on lines 116 to 118
HIPBLASLT_COMPUTE_F32_FAST_XF32 = 301, /**< XF32 compute for 32-bit input and output matrices */
HIPBLASLT_COMPUTE_F64 = 302, /**<32-bit floating-point precision.*/
HIPBLASLT_COMPUTE_F64 = 302, /**<64-bit floating-point precision.*/
HIPBLASLT_COMPUTE_I32 = 303,

Copy link
Copy Markdown
Contributor Author

Choose a reason for hiding this comment

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

A typo ?

Comment thread clients/gtest/matmul_gtest.yaml Resolved
@solaslin
solaslin marked this pull request as ready for review December 28, 2023 06:29

@jichangjichang jichangjichang left a comment

Copy link
Copy Markdown
Collaborator

Choose a reason for hiding this comment

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

lgtm

@KKyang
KKyang merged commit ed61c98 into ROCm:develop Jan 4, 2024
assistant-librarian Bot pushed a commit that referenced this pull request Jul 21, 2025
[hipblaslt] Adjust HIPBLASLT_TENSILE_LIBRARY_DIR on Windows. (#522)

This gets hipBLASLt tests passing on Windows in TheRock (see
ROCm/TheRock#783), for
ROCm/TheRock#681. It also enables PyTorch
builds on Windows to use hipBLASLt without needing to manually copy
folders around in the install directory.

This directory structure is now consistent with rocBLAS:
https://github.com/ROCm/rocm-libraries/blob/c7fe831061a4cdc3a14e1aacab8c0738c21daf32/projects/rocblas/library/src/CMakeLists.txt#L825-L830
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.

3 participants