-
Notifications
You must be signed in to change notification settings - Fork 50
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
Enabed Complex data type for Gemm #462
Enabed Complex data type for Gemm #462
Conversation
6943043
to
0c8ab21
Compare
369c21a
to
626d110
Compare
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
LGTM!
693e124
to
b53d0f6
Compare
@@ -186,5 +186,149 @@ struct Packetize { | |||
} | |||
}; | |||
|
|||
#ifdef BLAS_ENABLE_COMPLEX |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
I think it's best if we move this to a separate file to avoid confusion between the normal load/store.
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Moved the complex section to gemm_load_store_complex.hpp
} | ||
|
||
if (batch > 1 && batch_type == gemm_batch_type_t::interleaved) { |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
We could move this if condition to before the call to reference_blas
to avoid running the cgemm
unnecessarily.
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Yes done!
c8b7fbe
to
622318e
Compare
622318e
to
6a0e010
Compare
19b0fed
into
codeplaysoftware:master
This PR extends support to sycl::complex<float/double> data types to GEMM operators (gemm & batched extensions).
Main changes :
ext/oneapi/experimental/sycl_complex.hpp