[Ascend] Wx/reimpl multinomial op (#1218)

* reimpl multinomial op

Author: POI-WX

impl/ascend/functions/multinomial.cpp

@@ -4,21 +4,15 @@
  * @copyright  (c) 2023, DeepLink.
  */
 
+#include "../aclnn/adaptor.hpp"
 #include "../common/acloprunner.hpp"
 
 namespace impl {
 namespace ascend {
 diopiError_t diopiMultinomial(diopiContextHandle_t ctx, diopiTensorHandle_t out, diopiConstTensorHandle_t input, int64_t numSamples, bool replacement,
                               diopiGeneratorHandle_t generator) {
-    auto pair = getSeedAndOffset(ctx, generator, 10);
-    AclOpRunner<3, 1>("MultinomialWithReplacement", ctx)
-        .addInput(input)
-        .addConstInput(pair.first, diopi_dtype_int64)
-        .addConstInput(pair.second, diopi_dtype_int64)
-        .setAttr("numsamples", numSamples)
-        .setAttr("replacement", replacement)
-        .addOutput(out)
-        .run();
+    std::pair<uint64_t, int64_t> pair = getSeedAndOffset(ctx, generator, 10);
+    DIOPI_ASCEND_CALL_ACLNN(aclnnMultinomial, ctx, input, numSamples, replacement, static_cast<int64_t>(pair.first), pair.second, out);
     return diopiSuccess;
 }
 

impl/ascend_npu/ascend_config.yaml

@@ -154,6 +154,7 @@ ascend:
 - diopiMulInp
 - diopiMulInpScalar
 - diopiMulScalar
+- diopiMultinomial
 - diopiMin
 - diopiMinAll
 - diopiMinimum
@@ -235,9 +236,7 @@ ascend:
 - diopiZeros
 ascend_npu:
 - diopiNonzero
-- diopiCat
 - diopiCopyInp
-- diopiMultinomial
 - diopiRotaryEmbedding
 - diopiIndexPut
 - diopiIndexPutInp