SYCL: Add COUNT_EQUAL operator support#15991
Conversation
NeoZhangJianyu
left a comment
There was a problem hiding this comment.
- could you rebase the code since sycl_parallel_for() is reverted to stream->parallel_for() in base?
- same commit of #15937.
Thank you!
d401537 to
337d9f8
Compare
3031181 to
b39cc2d
Compare
|
|
||
| return a - b; | ||
|
|
||
| static __dpct_inline__ float op_count_equal(const float a, const float b) { |
There was a problem hiding this comment.
please remove duplicated code in this file.
There was a problem hiding this comment.
Thanks for the review!
I removed the duplicated op_count_equal definition from binbcast.hpp as requested.
The branch is now rebased on the latest master and clean.
NeoZhangJianyu
left a comment
There was a problem hiding this comment.
Thank you for great work!
3b29473 to
163b949
Compare
d3b3536 to
0f6b7fd
Compare
0f6b7fd to
b427efb
Compare
|
The iOS CI job is failing due to a missing SDK on the runner (iOS 26.0 is not installed), which seems unrelated to this PR. |
* SYCL: Add COUNT_EQUAL operator support (rebased on master) * SYCL: remove duplicate op_count_equal definition * tests: remove test_count_equal_typed and use test_count_equal for all cases * tests: keep only I32 case for COUNT_EQUAL as suggested * tests: keep only I32 case for COUNT_EQUAL as requested
* SYCL: Add COUNT_EQUAL operator support (rebased on master) * SYCL: remove duplicate op_count_equal definition * tests: remove test_count_equal_typed and use test_count_equal for all cases * tests: keep only I32 case for COUNT_EQUAL as suggested * tests: keep only I32 case for COUNT_EQUAL as requested
Summary
This PR adds GGML_OP_COUNT_EQUAL support to the SYCL backend.
Changes
ggml_sycl_op_count_equal()and its wrapperggml_sycl_count_equal()GGML_OP_COUNT_EQUALin:ggml_sycl_compute_forward()ggml_backend_sycl_device_supports_op()docs/ops.mdand regenerateddocs/ops/SYCL.csvto reflect SYCL supportF32,F16,I32, andI16Verification
cmake+make