[ILUVATAR_GPU] Support for iluvatar_gpu #16518
Open
+29
−2
Conversation
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
|
Thanks for your contribution! |
cuicheng01
reviewed
Sep 23, 2025
ppocr/data/imaug/operators.py
Outdated
| img, [ratio_h, ratio_w] = self.resize_image_type1(img) | ||
| data["image"] = img | ||
| data["shape"] = np.array([src_h, src_w, ratio_h, ratio_w]) | ||
| data["shape"] = np.array([src_h, src_w, ratio_h, ratio_w]).astype(np.float32) |
There was a problem hiding this comment.
天数卡上目前暂不支持double，这里不转的话在dataloader中会存在double的tensor。
感觉这里用fp32也可以，如果主逻辑需要使用fp64的话，我后面把这块另外搞个天数的分支。
There was a problem hiding this comment.
这里我不确定是否会对其他硬件造成影响，保守起见，可以单独出一个分支？
tianyuzhou668
force-pushed
the
iluvatar/0917
branch
from
aba8242 to
a063aee
Compare
tianyuzhou668
force-pushed
the
iluvatar/0917
branch
from
a063aee to
981a110
Compare
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
Add this suggestion to a batch that can be applied as a single commit.
This suggestion is invalid because no changes were made to the code.
Suggestions cannot be applied while the pull request is closed.
Suggestions cannot be applied while viewing a subset of changes.
Only one suggestion per line can be applied in a batch.
Add this suggestion to a batch that can be applied as a single commit.
Applying suggestions on deleted lines is not supported.
You must change the existing code in this line in order to create a valid suggestion.
Outdated suggestions cannot be applied.
This suggestion has been applied or marked resolved.
Suggestions cannot be applied from pending reviews.
Suggestions cannot be applied on multi-line comments.
Suggestions cannot be applied while the pull request is queued to merge.
Suggestion cannot be applied right now. Please check back later.
针对天数硬件做了适配