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

[Fix](bangc-ops): add large tensor num check for psamask. #818

Merged
merged 1 commit into from
Aug 24, 2023

Conversation

DanieeelLiu
Copy link
Collaborator

@DanieeelLiu DanieeelLiu commented Aug 23, 2023

Thanks for your contribution and we appreciate it a lot.

1. Motivation

add large tensor num check for psamask.

2. Modification

modified: bangc-ops/kernels/psamask/psamask.cpp

3. Test Report

3.1.4 Parameter Check

the test points are given and the test results are stated.

Test Point Acceptance Standard Test Result (Error Message)
Large tensor check
inputs : "shape":[1,1,2,2147483640]
Normal error [2023-8-23 15:5:6] [MLUOP] [Error]:[mluOpPsamaskForward] Overflow max tensor num. Currently, MLU-OPS supports tensor num smaller than 2^31.
[2023-8-23 15:5:6] [MLUOP] [Error]:"MLUOP_STATUS_BAD_PARAM in mluOpPsamaskForward(handle_, psa_type, input_desc, input, h_mask, w_mask, output_desc, output)"

3.4 Summary Analysis

Test passed

@PetrelYy PetrelYy added the BANGC label Aug 23, 2023
@PetrelYy PetrelYy added this to the v0.9.0 milestone Aug 23, 2023
@duzekunKTH duzekunKTH merged commit dd1f94f into Cambricon:master Aug 24, 2023
1 check 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.

4 participants