fix split (DeepLink-org#851)

* fix split

Author: hellozmz

impl/ascend/ascend_tensor.hpp

@@ -54,7 +54,7 @@ namespace ascend {
 inline void error(const char* file, int lineNum, const char* funcName, const char* format, ...) {
     va_list args;
     va_start(args, format);
-    printf("ERROR:[%s:%d in func:%s] : ", __FILE__, __LINE__, __FUNCTION__);
+    printf("ERROR:[%s:%d in func:%s] : ", file, lineNum, funcName);
     vprintf(format, args);
     printf("\n");
     throw std::runtime_error("error occuers");
@@ -63,15 +63,15 @@ inline void error(const char* file, int lineNum, const char* funcName, const cha
 inline void warning(const char* file, int lineNum, const char* funcName, const char* format, ...) {
     va_list args;
     va_start(args, format);
-    printf("WARNING:[%s:%d in func:%s]: ", __FILE__, __LINE__, __FUNCTION__);
+    printf("WARNING:[%s:%d in func:%s]: ", file, lineNum, funcName);
     vprintf(format, args);
     printf("\n");
 }
 
 inline void info(const char* file, int lineNum, const char* funcName, const char* format, ...) {
     va_list args;
     va_start(args, format);
-    printf("INFO:[%s:%d in func:%s]: ", __FILE__, __LINE__, __FUNCTION__);
+    printf("INFO:[%s:%d in func:%s]: ", file, lineNum, funcName);
     vprintf(format, args);
     printf("\n");
 }

impl/ascend/device_configs.py

@@ -662,18 +662,6 @@
         ),
     ),
 
-    'split': dict( # llm used
-        name=['split'],
-        tensor_para=dict(
-            args=[
-                {
-                    "ins": ['tensor'],
-                    "dtype": [Skip(np.float64)],
-                },
-            ]
-        ),
-    ),
-
     'transpose': dict(
         name=['transpose'],
         tensor_para=dict(

impl/ascend/functions/split.cpp

@@ -27,7 +27,7 @@ diopiError_t diopiSplitWithSizes(diopiContextHandle_t ctx, diopiTensorHandle_t*
     AclOpRunner<3, 1>("SplitV", ctx)
         .addInput(input)
         .addConstInput(splitSizes)
-        .addConstInput(dim, diopi_dtype_int64)
+        .addConstInput(dim, diopi_dtype_int32)
         .setAttr("num_split", numOuts)
         .addDynamicOutput(dynamicOutput, inputTensor.dtype())
         .run();