change more float32 to get_default_dtype

Author: HydrogenSulfate

ppsci/arch/physx_transformer.py

@@ -308,7 +308,9 @@ def _init_weights(self, module):
 
     def get_position_embed(self, x):
         B, N, _ = x.shape
-        position_ids = paddle.arange(0, N, dtype="float32").reshape([1, N, 1])
+        position_ids = paddle.arange(0, N, dtype=paddle.get_default_dtype()).reshape(
+            [1, N, 1]
+        )
         position_ids = position_ids.repeat_interleave(B, axis=0)
 
         position_embeds = paddle.zeros_like(x)

ppsci/utils/misc.py

@@ -148,7 +148,11 @@ def combine_array_with_time(x, t):
     nx = len(x)
     tx = []
     for ti in t:
-        tx.append(np.hstack((np.full([nx, 1], float(ti), dtype="float32"), x)))
+        tx.append(
+            np.hstack(
+                (np.full([nx, 1], float(ti), dtype=paddle.get_default_dtype()), x)
+            )
+        )
     tx = np.vstack(tx)
     return tx
 

ppsci/validate/geo_validator.py

@@ -20,6 +20,7 @@
 from typing import Union
 
 import numpy as np
+import paddle
 import sympy
 from sympy.parsing import sympy_parser as sp_parser
 from typing_extensions import Literal
@@ -152,7 +153,7 @@ def __init__(
                     label[key] = np.full(
                         (next(iter(input.values())).shape[0], 1),
                         label[key],
-                        "float32",
+                        paddle.get_default_dtype(),
                     )
             else:
                 raise NotImplementedError(f"type of {type(value)} is invalid yet.")

ppsci/visualize/vtu.py

@@ -63,7 +63,7 @@ def _save_vtu_from_array(filename, coord, value, value_keys, num_timestamp=1):
         if coord_ndim == 2:
             axis_x = np.ascontiguousarray(coord[t * nx : (t + 1) * nx, 0])
             axis_y = np.ascontiguousarray(coord[t * nx : (t + 1) * nx, 1])
-            axis_z = np.zeros([nx], dtype="float32")
+            axis_z = np.zeros([nx], dtype=paddle.get_default_dtype())
         elif coord_ndim == 3:
             axis_x = np.ascontiguousarray(coord[t * nx : (t + 1) * nx, 0])
             axis_y = np.ascontiguousarray(coord[t * nx : (t + 1) * nx, 1])