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

Add Support to perform complex f32 f64 contractions with complex compute types #174

Merged
merged 14 commits into from
Jan 4, 2024

Conversation

mkarunan
Copy link
Contributor

Add Device class specialization for complex types
Add samples to test complex float 32 contraction
Add pack, unpack utils to perform data pre and post processing

@mkarunan mkarunan marked this pull request as ready for review December 27, 2023 18:30
@CongMa13 CongMa13 force-pushed the add-complex-f32-f64 branch from 88e6cbf to b2856df Compare January 2, 2024 21:06
mkarunan and others added 14 commits January 4, 2024 00:18
- Add API changes
- Add test files
- Add samples
- Add cpu contraction for complex types
- Add complex instances
- Add compute types
- Add instance factory for scale
- Modify complex scale impl
- Modify pack func
- Remove intermediate buffers
- Add cpu instances
- Add scale complex initial imp
- Remove compute f32 instances for complex f32
- Modify multiply utility
- Modified bilinear to take complex compute
- Fix issue in yaml_test
- Fix a bug in hiptensorInitContractionDescriptor
- Fix bug in samples
- Add unit tests for scale contraction
- Add complex_scale_contraction_test.cpp
- Fixed bug in device_contraction_bilinear_complex.hpp
- New single kernel selection. To be improved.
- Used instance selected by brute force to compute tensor with 1 as the
most right stride
- Fixed bug that used data type id as compute data type id
@CongMa13 CongMa13 force-pushed the add-complex-f32-f64 branch from b2856df to 348e281 Compare January 4, 2024 00:25
Copy link
Collaborator

@cgmillette cgmillette left a comment

Choose a reason for hiding this comment

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

LGTM

@CongMa13 CongMa13 merged commit edf1627 into ROCm:develop Jan 4, 2024
3 of 9 checks passed
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