Skip to content

[Metax] fix cufft and fix some blas kernel apply #1976

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

Open
metax666 wants to merge 12 commits into
base: develop
Choose a base branch
from
Open

[Metax] fix cufft and fix some blas kernel apply #1976

Changes from all commits
Commits
Show all changes
12 commits
Select commit Hold shift + click to select a range
69f3721
[fix] fix fail test when backend is mack
zhang-chenyi Sep 4, 2025
e45d324
[Metax] fix fail test when backend is mack
metax666 Sep 4, 2025
ef9d554
Merge branch 'PaddlePaddle:develop' into develop
metax666 Sep 4, 2025
a1530d2
[metax]change_cupti_and_fix_softmax (#7)
duqimeng Sep 9, 2025
352f02e
[Metax] fix dgc & mklml compile product path problem (#8)
StareAtYou Sep 9, 2025
8f13fae
[Metax] fix accuracy kernel & add test_accuracy_op_metax.py unit test…
StareAtYou Sep 11, 2025
8938293
[Metax] update metax_gpu CMakeLists.txt (#10)
StareAtYou Sep 11, 2025
f54187f
[metax] updata_qr_kernel (#11)
duqimeng Sep 11, 2025
7964c35
Merge branch 'PaddlePaddle:develop' into develop
metax666 Sep 12, 2025
1e04216
[Metax] fix illegal address access error in test_momentum_op (#12)
StareAtYou Sep 15, 2025
aca80a4
[Metax] fix cufft and fix some blas kernel apply (#13)
duqimeng Sep 15, 2025
e595708
Revert "[Metax] fix cufft and fix some blas kernel apply (#13)"
metax666 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