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.

Sign up for GitHub

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

Jump to bottom

Copy wmma type files #154

Closed
wants to merge 5 commits into from
Closed

Copy wmma type files #154

wants to merge 5 commits into from

Conversation

CongMa13
Copy link
Collaborator

  1. Copied xfloat32 and type_trait, type_ext files from wmma to hiptensor.
  2. Use type_trait of f16 to replace hard-code number.

@CongMa13
Add CPU implementation of elementwise permutation
074c000 
- Add CPU implementation of elementwise permutation -  permutationByCpu
- Add sample of elementwise permutation

GPU implementation of elementwise permutation is not done yet.
@CongMa13
Move files used by all tests out of 01_contraction
60f4b90 
moved 01_contraction/device/common.hpp -> device/common.hpp
renamed 01_contraction/common.hpp -> utils.hpp since there is already a
file named common.hpp in test folder
@CongMa13
Implement permutation with CK
bf1c7ee 
- implemented `hiptensorPermutation` with CK
- added unit tests for `hiptensorPermutation`
@CongMa13
Support both row_major and col_major data layout
082a966 
Data layout of cuTenor is col_major by default. HipTensor contraction
only supports row_major for now. HipTensor supports both col_major
and row_major to make users be able to choose which layout they want to
use.

- Use CMake option `HIPTENSOR_DATA_LAYOUT_COL_MAJOR` to choose layout
@CongMa13
Add type related files
00fac42 
new file:   include/native_types.hpp
new file:   include/native_types_impl.hpp
new file:   include/type_traits.hpp
new file:   include/types_ext.hpp
new file:   include/xfloat32.hpp
@CongMa13 CongMa13 closed this Nov 17, 2023
@CongMa13 CongMa13 deleted the copy_wmma_type_files branch May 3, 2024 20:25
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Reviewers

@cgmillette cgmillette Awaiting requested review from cgmillette cgmillette will be requested when the pull request is marked ready for review cgmillette is a code owner

@bragadeesh bragadeesh Awaiting requested review from bragadeesh bragadeesh will be requested when the pull request is marked ready for review bragadeesh is a code owner

@dlangbe dlangbe Awaiting requested review from dlangbe dlangbe will be requested when the pull request is marked ready for review dlangbe is a code owner

@afanfa afanfa Awaiting requested review from afanfa afanfa will be requested when the pull request is marked ready for review afanfa is a code owner

Assignees
No one assigned
Labels
None yet
Projects
None yet
Milestone
No milestone
Development

Successfully merging this pull request may close these issues.
1 participant
@CongMa13