Skip to content
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

[Enhancement] Support BoolTensor and LongTensor on Ascend NPU #1011

Merged
merged 1 commit into from Apr 10, 2023

Conversation

Ginray
Copy link
Contributor

@Ginray Ginray commented Mar 24, 2023

Motivation

When we test the ssd model on the NPU, an assert error occurs during the eval phase, because npu.BoolTensor and npu.LongTensor are not supported in instane_data.py.

Modification

Modify the instane_data.py file to support NPU BoolTensor and LongTensor.

BC-breaking (Optional)

Not affected.

Use cases (Optional)

Not affected.

@CLAassistant
Copy link

CLAassistant commented Mar 24, 2023

CLA assistant check
All committers have signed the CLA.

@Ginray Ginray force-pushed the dev-for-tensor-fixed branch 2 times, most recently from fbe04b7 to 67d513b Compare March 24, 2023 09:20
mmengine/structures/instance_data.py Outdated Show resolved Hide resolved
mmengine/structures/instance_data.py Outdated Show resolved Hide resolved
mmengine/structures/instance_data.py Show resolved Hide resolved
HAOCHENYE
HAOCHENYE previously approved these changes Apr 6, 2023
@zhouzaida zhouzaida changed the title [Fix] Support BoolTensor and LongTensor on Ascend NPU. [Fix] Support BoolTensor and LongTensor on Ascend NPU Apr 10, 2023
@zhouzaida zhouzaida changed the title [Fix] Support BoolTensor and LongTensor on Ascend NPU [Enhancement] Support BoolTensor and LongTensor on Ascend NPU Apr 10, 2023
@zhouzaida zhouzaida merged commit 1c67f9e into open-mmlab:main Apr 10, 2023
16 of 19 checks passed
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

None yet

4 participants