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

update OneMKL gemm_batch call inside dpnp.matmul and column_major version of gemm #1793

Merged
merged 13 commits into from
May 8, 2024
+589 −397
Merged

update OneMKL gemm_batch call inside dpnp.matmul and column_major version of gemm #1793

Changes from 1 commit
Commits
Show all changes
13 commits
Select commit Hold shift + click to select a range
f12d833
updating gemm_batch call
vtavana Apr 15, 2024
78bec82
fix failed tests for dot function
vtavana Apr 19, 2024
00a7755
Merge branch 'master' into update-gemm-batch
antonwolfy Apr 22, 2024
0da5437
address comments
vtavana Apr 24, 2024
c5141c9
Merge branch 'master' into update-gemm-batch
vtavana Apr 24, 2024
f7d7527
Merge branch 'master' into update-gemm-batch
vtavana May 3, 2024
2275290
split batch_size and column major
vtavana Apr 25, 2024
d9aa16e
fix failed tests
vtavana May 5, 2024
e72c85f
simplify gemm implementation
vtavana May 7, 2024
6897b98
Merge branch 'master' into update-gemm-batch
vtavana May 7, 2024
fa7cc33
waiting for host task before changing results
vtavana May 7, 2024
6cce243
address comments
vtavana May 8, 2024
db21604
reorder functions in dpnp/dpnp_utils/dpnp_utils_linearalgebra.py to k…
vtavana May 8, 2024
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
4 changes: 2 additions & 2 deletions dpnp/dpnp_utils/dpnp_utils_linearalgebra.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -2063,6 +2063,7 @@ def dpnp_matmul(
dep_events_list,
)
host_tasks_list.append(ht_blas_ev)
dpctl.SyclEvent.wait_for(host_tasks_list)
vtavana marked this conversation as resolved.
Show resolved Hide resolved
if not row_major:
# TODO: investigate the possibility of defining result
# array with "F" order for this case
Expand All @@ -2077,8 +2078,7 @@ def dpnp_matmul(
result,
dep_events_list,
)

dpctl.SyclEvent.wait_for(host_tasks_list)
dpctl.SyclEvent.wait_for(host_tasks_list)

if appended_axes:
result = dpnp.squeeze(result, tuple(appended_axes))
Expand Down
Loading