Skip to content

[Metax] fix illegal address access error in test_momentum_op #12

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

Merged
merged 10 commits into from
Sep 15, 2025
Merged

[Metax] fix illegal address access error in test_momentum_op #12

Changes from all commits
Commits
Show all changes
10 commits
Select commit Hold shift + click to select a range
055d00f
[Metax] fix dgc & mklml compile product path problem
StareAtYou Sep 9, 2025
0021e7e
Merge branch 'metax666:develop' into develop
StareAtYou Sep 9, 2025
ef96403
[Metax] fix accuracy kernel & add test_accuracy_op_metax.py unit test
StareAtYou Sep 11, 2025
5ec97b8
[Metax] add mixed_vector fix & update change patch
StareAtYou Sep 11, 2025
fa31211
[Metax] update metax_gpu CMakeLists.txt
StareAtYou Sep 11, 2025
543ee63
Merge branch 'metax666:develop' into develop
StareAtYou Sep 11, 2025
9364073
Merge branch 'metax666:develop' into develop
StareAtYou Sep 11, 2025
3b77cb0
Merge branch 'metax666:develop' into develop
StareAtYou Sep 12, 2025
7814d95
Merge branch 'metax666:develop' into develop
StareAtYou Sep 15, 2025
9066beb
[Metax] fix illegal address access error in test_momentum_op
StareAtYou Sep 15, 2025
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
6 changes: 4 additions & 2 deletions backends/metax_gpu/patch/tmp/mixed_vector.h
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -386,7 +386,8 @@ class MixVector {

// the unify method to access CPU or CUDA data. immutable.
const T *Data(phi::Place place) const {
if (place.GetType() == phi::AllocationType::GPU) {
if (place.GetType() == phi::AllocationType::GPU ||
place.GetType() == phi::AllocationType::CUSTOM) {
return CUDAData(place);
} else {
return data();
Expand All @@ -395,7 +396,8 @@ class MixVector {

// the unify method to access CPU or CUDA data. mutable.
T *MutableData(phi::Place place) {
if (place.GetType() == phi::AllocationType::GPU) {
if (place.GetType() == phi::AllocationType::GPU ||
place.GetType() == phi::AllocationType::CUSTOM) {
return CUDAMutableData(place);
} else {
return data();
Expand Down