diff --git a/paddle/fluid/eager/auto_code_generator/final_state_generator/eager_gen.py b/paddle/fluid/eager/auto_code_generator/final_state_generator/eager_gen.py index d1e208541537c8eddf69862aaabbf5fcf6d2a009..81d0c9b7bed598cae930fca0ff050a1c564ac306 100644 --- a/paddle/fluid/eager/auto_code_generator/final_state_generator/eager_gen.py +++ b/paddle/fluid/eager/auto_code_generator/final_state_generator/eager_gen.py @@ -25,10 +25,10 @@ core_ops_args_type_info = {} yaml_types_mapping = { - 'int' : 'int', 'int32_t' : 'int32_t', 'int64_t' : 'int64_t', 'size_t' : 'size_t', \ + 'int' : 'int', 'int32' : 'int32_t', 'int64' : 'int64_t', 'size_t' : 'size_t', \ 'float' : 'float', 'double' : 'double', 'bool' : 'bool', \ 'Backend' : 'paddle::experimental::Backend', 'DataLayout' : 'paddle::experimental::DataLayout', 'DataType' : 'paddle::experimental::DataType', \ - 'int64_t[]' : 'std::vector', 'int[]' : 'std::vector', + 'int64[]' : 'std::vector', 'int[]' : 'std::vector', 'Tensor' : 'Tensor', 'Tensor[]' : 'std::vector', 'Tensor[Tensor[]]' : 'std::vector>', diff --git a/python/paddle/utils/code_gen/api.yaml b/python/paddle/utils/code_gen/api.yaml index 699e42f23732a00d980d5268b5b7c454ae202b2f..8c68ca4d7e0e4666619d21fea7ec093ab49a666a 100644 --- a/python/paddle/utils/code_gen/api.yaml +++ b/python/paddle/utils/code_gen/api.yaml @@ -121,7 +121,7 @@ backward : matmul_grad - api : mean - args : (Tensor x, int64_t[] axis={}, bool keep_dim=false) + args : (Tensor x, int64[] axis={}, bool keep_dim=false) output : Tensor infer_meta : func : MeanInferMeta @@ -181,7 +181,7 @@ func : subtract - api : sum - args : (Tensor x, int64_t[] axis={}, DataType dtype=DataType::UNDEFINED, bool keep_dim=false) + args : (Tensor x, int64[] axis={}, DataType dtype=DataType::UNDEFINED, bool keep_dim=false) output : Tensor infer_meta : func : SumInferMeta diff --git a/python/paddle/utils/code_gen/api_base.py b/python/paddle/utils/code_gen/api_base.py index 601248a41763989df5e0324f16180ce2df2314dd..68127fb522c3dee7219ac244c88c42abdd5884d1 100644 --- a/python/paddle/utils/code_gen/api_base.py +++ b/python/paddle/utils/code_gen/api_base.py @@ -89,18 +89,20 @@ class BaseAPI(object): attr_types_map = { 'ScalarArray': 'const ScalarArray&', 'Scalar': 'const Scalar&', + 'uint8': 'uint8_t', 'int': 'int', - 'int32_t': 'int32_t', - 'int64_t': 'int64_t', + 'int32': 'int32_t', + 'int64': 'int64_t', 'long': 'long', 'size_t': 'size_t', 'float': 'float', 'double': 'double', 'bool': 'bool', + 'str': 'const std::string&', 'Backend': 'Backend', 'DataLayout': 'DataLayout', 'DataType': 'DataType', - 'int64_t[]': 'const std::vector&', + 'int64[]': 'const std::vector&', 'int[]': 'const std::vector&', 'long[]': 'const std::vector&' } @@ -110,8 +112,8 @@ class BaseAPI(object): 'ScalarArray': 'const paddle::optional&', 'Scalar': 'const paddle::optional&', 'int': 'paddle::optional', - 'int32_t': 'paddle::optional', - 'int64_t': 'paddle::optional', + 'int32': 'paddle::optional', + 'int64': 'paddle::optional', 'size_t': 'paddle::optional', 'float': 'paddle::optional', 'double': 'paddle::optional', @@ -119,7 +121,7 @@ class BaseAPI(object): 'Backend': 'paddle::optional', 'DataLayout': 'paddle::optional', 'DataType': 'paddle::optional', - 'int64_t[]': 'paddle::optional>', + 'int64[]': 'paddle::optional>', 'int[]': 'paddle::optional>' } diff --git a/python/paddle/utils/code_gen/sparse_api.yaml b/python/paddle/utils/code_gen/sparse_api.yaml index 135989121cca695b0e629192774af0eb3e41c812..b531c2ed9ce51a28c5ea43a205ed439e9dd5b328 100644 --- a/python/paddle/utils/code_gen/sparse_api.yaml +++ b/python/paddle/utils/code_gen/sparse_api.yaml @@ -11,7 +11,7 @@ invoke : to_dense_impl(x, backend) - sparse_api : to_sparse_coo - args : (Tensor x, Backend backend, int64_t sparse_dim) + args : (Tensor x, Backend backend, int64 sparse_dim) output : Tensor(out@SparseCooTensor) invoke : to_sparse_coo_impl(x, backend, sparse_dim)