wq/impl atan operator (DeepLink-org#812)

* wq/impl atan operator

Author: wanfengcxz

impl/ascend/convert_config.yaml

@@ -276,11 +276,17 @@
 - diopiRemainder:
     dtype: (bool, uint8, int8, int16, uint16)->int32, (float64)->float32
 
+- diopiAtan:
+    dtype: (uint8, int8, int32, int16, int64, bool)->float32
+
+- diopiAtanInp:
+    dtype: (uint8, int8, int32, int16, int64, bool)->float32
+
 - diopiNormalTensor:
     dtype: (float64)->float32
 
 - diopiNormalScalarTensor:
     dtype: (float64)->float32
 
 - diopiNormalTensorScalar:
-    dtype: (float64)->float32
+    dtype: (float64)->float32

impl/ascend/device_configs.py

@@ -338,7 +338,7 @@
     ),
 
     'pointwise_op': dict(
-        name=['erf', 'erfinv', 'asin', 'ceil', 'atan'],
+        name=['erf', 'erfinv', 'asin', 'ceil'],
         tensor_para=dict(
             args=[
                 {
@@ -350,7 +350,7 @@
     ),
 
     'pointwise_op_int_without_inplace': dict(
-        name=['erf', 'asin', 'atan'],
+        name=['erf', 'asin'],
         tensor_para=dict(
             args=[
                 {
@@ -362,7 +362,7 @@
     ),
 
     'pointwise_op_uint8': dict(
-        name=['erf', 'asin', 'atan'],
+        name=['erf', 'asin'],
         tensor_para=dict(
             args=[
                 {
@@ -374,7 +374,7 @@
     ),
 
     'pointwise_op_bool': dict(
-        name=['erf', 'asin', 'atan'],
+        name=['erf', 'asin'],
         tensor_para=dict(
             args=[
                 {

impl/ascend_npu/ascend_config.yaml

@@ -223,3 +223,5 @@ ascend_npu:
 - diopiTokenSoftmaxReduceVInference
 - diopiApplyPenalty
 - diopiContextAttentionInference
+- diopiAtan
+- diopiAtanInp

impl/ascend_npu/diopi_impl/atan.cpp

@@ -0,0 +1,30 @@
+/**
+ * @file
+ * @author DeepLink
+ * @copyright  (c) 2023, DeepLink.
+ */
+
+#include "helper.hpp"
+#include "op_plugin/AclOpsInterface.h"
+
+namespace OP_IMPL_NS {
+
+diopiError_t diopiAtan(diopiContextHandle_t ctx, diopiTensorHandle_t out, diopiConstTensorHandle_t input) {
+    BEGIN_CALL_ACL_OP(input, out);
+    if (!outAt.defined() || outAt.numel() <= 0) {
+        return diopiSuccess;
+    }
+    acl_op::atan_out(inputAt, outAt);
+    END_CALL_ACL_OP();
+}
+
+diopiError_t diopiAtanInp(diopiContextHandle_t ctx, diopiTensorHandle_t input) {
+    BEGIN_CALL_ACL_OP(input);
+    if (!inputAt.defined() || inputAt.numel() <= 0) {
+        return diopiSuccess;
+    }
+    acl_op::atan_(inputAt);
+    END_CALL_ACL_OP();
+}
+
+}  // namespace OP_IMPL_NS

impl/ascend_npu/diopi_impl/helper.hpp

@@ -326,7 +326,7 @@ template <>
 inline std::string dumpArgs(const at::IntArrayRef& t) {
     std::stringstream stream;
     stream << "[";
-    for (size_t i = 0; i < t.size(); i++) {
+    for (long i : t) {
         stream << i << ",";
     }
     stream << "]";