fix: use y.device instead of hardcoded .cuda() in data2vec compute_var - #3710

Open

li-lizhe wants to merge 3 commits into

modelscope:mainfrom

li-lizhe:fix/data2vec-compute-var-device

Commits on Sep 15, 2026

fix: use y.device instead of hardcoded .cuda() in data2vec compute_var
li-lizhe
committed
9851a35
test: add CPU/Gloo regression tests for data2vec compute_var
li-lizhe
committed
18ed7f1

Commits on Sep 16, 2026

fix(test): resolve compute_var import and f32_worker spawn pickle
li-lizhe
committed
944d3d9