-
Notifications
You must be signed in to change notification settings - Fork 57
Feature/add mul python api #548
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.
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
base: main
Are you sure you want to change the base?
Feature/add mul python api #548
Conversation
python/infinicore/ops/mul.py
Outdated
| import numpy as np | ||
|
|
||
|
|
||
| def _scalar_to_tensor(scalar_value, reference_tensor): |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
torch是这么做的?
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
我让他,先看看怎么实现这个功能
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
先去掉了这部分的支持,re-force push了
python/infinicore/ops/mul.py
Outdated
| """Convert a scalar value to a tensor with the same shape and dtype as reference_tensor""" | ||
| # Use torch to create the tensor first (handles dtype conversion correctly), | ||
| # then convert to infinicore tensor via from_blob | ||
| import torch |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
依赖torch?
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
是的,我先去掉这部分scalar的支持
869953e to
222b774
Compare
222b774 to
5c89a40
Compare
Uh oh!
There was an error while loading. Please reload this page.