fix fill inplace call for aclnn

DeepLink-org · Jun 13, 2024 · 1f0b8a7 · 1f0b8a7
1 parent f6baaf8
commit 1f0b8a7
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/impl/ascend/functions/nlllossv2.cpp b/impl/ascend/functions/nlllossv2.cpp
@@ -18,7 +18,8 @@ diopiError_t diopiNLLLossV2(diopiContextHandle_t ctx, diopiTensorHandle_t out, d
     AscendTensor inputAt(input);
     if (inputAt.numel() <= 0) {
         if (diopiReduction_t::ReductionMean == reduction) {
-            DIOPI_ASCEND_CALL_ACLNN(aclnnInpalceFillScalar, ctx, out, std::nanf(""));
+            diopiScalar_t nans{diopi_dtype_float64, std::nanf("")};
+            DIOPI_ASCEND_CALL_ACLNN(aclnnInplaceFillScalar, ctx, out, &nans);
         } else if (diopiReduction_t::ReductionSum == reduction || diopiReduction_t::ReductionNone == reduction) {
             DIOPI_ASCEND_CALL_ACLNN(aclnnInplaceZero, ctx, out);
         }