[bug] mf联调发现，参数初始化时，dim_map[shard_dim] index越界

Open Beginner friendly
#711 0 comments 0 reactions 0 assignees View on GitHub

Nobody has claimed this yet.

Assessment

Difficulty
2/5
Estimated time
1-3 hours
Newbie friendliness
72/100
Issue type
Bug
Clarity
Clearly specified
Activity status
Active
Tech stack
python
Domain
distributed-systems

Research direction

Start at DTensor parameter initialization and trace where dim_map is constructed from the tensor shape and device mesh dimensions. Confirm that the tensor shape's ndim is used, then verify that dim_map[shard_dim] no longer raises an index error during initialization.

Written by the indexing model from the issue text.

Description

问题：

DTensor 参数初始化时，发现dim_map[shard_dim] index越界。

原因：

dim_map 元祖构建的dim维度取错了，期望是Tensor shape的ndim，实际取成了devicemesh的ndim

解决：

改成shape的dim
image.png

schema_version: 1
source: gitcode
gitcode_repo: mindspore/hyper-parallel
gitcode_issue: 169
source_url: https://gitcode.com/mindspore/hyper-parallel/issues/169

Dominant language
Python
Stars
53
Forks
63
Avg merge
1d 4h
Merged PRs (30d)
76

Contributor guide

No contributing guide indexed for this repository

First steps

  1. Read the whole issue, then the project's contributing guide.
  2. Comment on the issue to say you are picking it up — it saves two people doing the same work.
  3. Fork the repository and make your change on a branch.
  4. Open a pull request that references the issue number.

More from mindspore-ai/hyper-parallel

All issues in mindspore-ai/hyper-parallel

Similar issues

More Python issues

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.