]> git.djapps.eu Git - pkg/ggml/sources/ggml/commit
CANN: fix acl_rstd allocation size in ggml_cann_rms_norm (llama/15760)
authorChenguang Li <redacted>
Thu, 4 Sep 2025 03:03:02 +0000 (11:03 +0800)
committerGeorgi Gerganov <redacted>
Fri, 5 Sep 2025 09:54:13 +0000 (12:54 +0300)
commitfd4a4653f81e1ac317103f8492f51575eb49b68b
tree8816ca851f99a967771ff20e4bd1675484fb1cbe
parent2600727943e2587cb447d42c0ad278a26a23b779
CANN: fix acl_rstd allocation size in ggml_cann_rms_norm (llama/15760)

Fixes #15330

Adjust the allocation size of acl_rstd. The parameter `dims` is set to 3 according to the CANN documentation.

Co-authored-by: Yuchuan <redacted>
src/ggml-cann/aclnn_ops.cpp