关于GCU、沐曦GPGPU、MLU、0卡V100资源4月7日恢复上架的公告>>> 关于共建具身智能开源数据集的倡议>>> 关于云脑任务中统一路径访问方式的公告>>> 关于将启智集群GPU资源迁移至智算集群的公告>>>
OpenI
/
MSAdapter
Not watched
18
38
Fork 65
Code
Releases 5 Wiki Activity
Issues 2 Pull Requests 8 Datasets Model Cloudbrain
Pull Requests
/
Detail
Labels Milestones
New Pull Request

#182 add torch.ne and tensor.expand() on GPU.

Merged
zoulq merged 1 commits from frelam/MSAdapter:master20221122-2 into master 1 year ago
Conversation 0 Commits 1 Files Changed 3
frelam commented 1 year ago
zoulq reviewed 1 year ago
ms_adapter/pytorch/tensor.py
@@ -430,2 +430,4 @@
input_ms = ms.ops.cast(input_ms, mstype.int32)
_size = size_to_ms_tensor(size)
# TODO: ms.ops.expand() to support on GPU and delete 'broadcast_to' code.
if get_backend() == 'GPU':
zoulq commented 1 year ago

可以直接用is_under_gpu_context

可以直接用is_under_gpu_context
frelam commented 1 year ago

done

done
zoulq reviewed 1 year ago
ms_adapter/utils.py
@@ -22,2 +23,4 @@
def is_under_gpu_context():
return get_backend() == 'GPU'

@constexpr
zoulq commented 1 year ago

不需要加这个，后续会封装接口统一整改

不需要加这个，后续会封装接口统一整改
frelam marked this conversation as resolved
zoulq merged commit 88bdcf7465 into master 1 year ago
The pull request has been merged as 88bdcf7465.
Sign in to join this conversation.
Reviewers
Request review
No reviewers
Labels
Apply labels
Clear labels
No items
No Label
Milestone
Set milestone
Clear milestone
No items
No Milestone
Assignees
Assign users
Clear assignees
No Assignees
2 Participants
Notifications
Due Date

No due date set.

Dependencies

This pull request currently doesn't have any dependencies.

Write Preview
Loading…
Cancel
Save
There is no content yet.
Delete Branch 'frelam/MSAdapter:master20221122-2'

Deleting a branch is permanent. It CANNOT be undone. Continue?

No
Yes
Please read the following content carefully:

Dear OpenI User

Thank you for your continuous support to the Openl Qizhi Community AI Collaboration Platform. In order to protect your usage rights and ensure network security, we updated the Openl Qizhi Community AI Collaboration Platform Usage Agreement in January 2024. The updated agreement specifies that users are prohibited from using intranet penetration tools. After you click "Agree and continue", you can continue to use our services. Thank you for your cooperation and understanding.

For more agreement content, please refer to the《Openl Qizhi Community AI Collaboration Platform Usage Agreement》

Agree and continue
Disagree, exit