Skip to content

fix(ascend): fix sum global reduce, add abs/acos ops, fix nll_loss ou… - #60

Merged
Hchnr merged 1 commit into
mainfrom
ascend
May 27, 2026
Merged

fix(ascend): fix sum global reduce, add abs/acos ops, fix nll_loss ou…#60
Hchnr merged 1 commit into
mainfrom
ascend

Conversation

@Hchnr

@Hchnr Hchnr commented May 27, 2026

Copy link
Copy Markdown
Owner

…tput shape

  • sum.cc: fix global reduce (no dim or empty dim) passing empty array to ACL causing ret=161002; now fills all dims automatically
  • abs.cc: add abs op Ascend backend implementation
  • acos.cc: add acos op Ascend backend with float64->float32 downcast
  • op_api_common.h: add kDouble -> ACL_DOUBLE mapping in ToAclDataType
  • nll_loss.cc: fix output/total_weight scalar shape causing ACL ret=161001; use shape{1} then squeeze back to scalar
  • backends_ascend.conf: register abs, acos to ascend backend
  • test_acos_dispatch.py: change float64 param to float16 (device rand fp64 bug)

…tput shape

- sum.cc: fix global reduce (no dim or empty dim) passing empty array to ACL
  causing ret=161002; now fills all dims automatically
- abs.cc: add abs op Ascend backend implementation
- acos.cc: add acos op Ascend backend with float64->float32 downcast
- op_api_common.h: add kDouble -> ACL_DOUBLE mapping in ToAclDataType
- nll_loss.cc: fix output/total_weight scalar shape causing ACL ret=161001;
  use shape{1} then squeeze back to scalar
- backends_ascend.conf: register abs, acos to ascend backend
- test_acos_dispatch.py: change float64 param to float16 (device rand fp64 bug)
@Hchnr
Hchnr merged commit fba07c9 into main May 27, 2026
1 check failed
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.

1 participant