diff --git a/actx_dgfem_suite/codegen/pytato_target.py b/actx_dgfem_suite/codegen/pytato_target.py index a5d1db6..abe81bd 100644 --- a/actx_dgfem_suite/codegen/pytato_target.py +++ b/actx_dgfem_suite/codegen/pytato_target.py @@ -311,8 +311,7 @@ def rec(self, expr: ArrayOrNames) -> str: elts=[ self._get_tag_expr(tag) for tag in expr.tags - if tag.__class__.__name__ - not in ["NameHint"] + if tag.__class__.__name__ not in ["NameHint"] ] ), ast.Name(lhs), diff --git a/actx_dgfem_suite/suite/tiny_maxwell_3D_P1/literals.npz.dvc b/actx_dgfem_suite/suite/tiny_maxwell_3D_P1/literals.npz.dvc index 6e5e846..e26a8a4 100644 --- a/actx_dgfem_suite/suite/tiny_maxwell_3D_P1/literals.npz.dvc +++ b/actx_dgfem_suite/suite/tiny_maxwell_3D_P1/literals.npz.dvc @@ -1,5 +1,5 @@ outs: -- md5: 7af3667004d41dc033fc60ff43497e35 - size: 119240 +- md5: 780cef037c7682838ed452ecd3b8d410 + size: 118856 hash: md5 path: literals.npz diff --git a/actx_dgfem_suite/suite/tiny_maxwell_3D_P1/main.py b/actx_dgfem_suite/suite/tiny_maxwell_3D_P1/main.py index 311965d..226132e 100644 --- a/actx_dgfem_suite/suite/tiny_maxwell_3D_P1/main.py +++ b/actx_dgfem_suite/suite/tiny_maxwell_3D_P1/main.py @@ -58,62 +58,15 @@ def _rhs_inner( ], ) _pt_t_unit_10 = make_loopy_program( - "{ [_0, _1] : 0 <= _0 <= 191 and 0 <= _1 <= 2 }", - "out[_0, _1] = in_0[in_1[_0, 0] % 384, in_2[_0, _1] % 4]", - [ - lp.GlobalArg("out", dtype=np.float64, shape=(192, 3)), - lp.GlobalArg("in_0", dtype=np.float64, shape=(384, 4)), - lp.GlobalArg("in_1", dtype=np.int32, shape=(192, 1)), - lp.GlobalArg("in_2", dtype=np.int64, shape=(192, 3)), - ], - ) - _pt_t_unit_11 = make_loopy_program( - "{ [_0, _1] : 0 <= _0 <= 191 and 0 <= _1 <= 2 }", - "out[_0, _1] = in_0[in_1[_0, 0] % 384, in_2[_0, _1] % 4]", - [ - lp.GlobalArg("out", dtype=np.complex128, shape=(192, 3)), - lp.GlobalArg("in_0", dtype=np.complex128, shape=(384, 4)), - lp.GlobalArg("in_1", dtype=np.int32, shape=(192, 1)), - lp.GlobalArg("in_2", dtype=np.int64, shape=(192, 3)), - ], - ) - _pt_t_unit_12 = make_loopy_program( - "{ [_0, _1] : 0 <= _0 <= 191 and 0 <= _1 <= 2 }", - "out[_0, _1] = cast(np:dtype('complex128'), _in0[_0, 0])*_in1[_0, _1]", - [ - lp.GlobalArg("out", dtype=np.complex128, shape=(192, 3)), - lp.GlobalArg("_in0", dtype=np.float64, shape=(192, 1)), - lp.GlobalArg("_in1", dtype=np.complex128, shape=(192, 3)), - ], - ) - _pt_t_unit_13 = make_loopy_program( "{ [_0, _1] : 0 <= _0 <= 1535 and 0 <= _1 <= 2 }", "out[_0, _1] = in_0[in_1[_0, 0] % 192, in_2[_0, _1] % 3]", [ - lp.GlobalArg("out", dtype=np.complex128, shape=(1536, 3)), - lp.GlobalArg("in_0", dtype=np.complex128, shape=(192, 3)), + lp.GlobalArg("out", dtype=np.float64, shape=(1536, 3)), + lp.GlobalArg("in_0", dtype=np.float64, shape=(192, 3)), lp.GlobalArg("in_1", dtype=np.int32, shape=(1536, 1)), lp.GlobalArg("in_2", dtype=np.int64, shape=(1536, 3)), ], ) - _pt_t_unit_14 = make_loopy_program( - "{ [_0, _1] : 0 <= _0 <= 1343 and 0 <= _1 <= 2 }", - "out[_0, _1] = _in0[_0, 0]*_in1[_0, _1]", - [ - lp.GlobalArg("out", dtype=np.float64, shape=(1344, 3)), - lp.GlobalArg("_in0", dtype=np.float64, shape=(1344, 1)), - lp.GlobalArg("_in1", dtype=np.float64, shape=(1344, 3)), - ], - ) - _pt_t_unit_15 = make_loopy_program( - "{ [_0, _1] : 0 <= _0 <= 191 and 0 <= _1 <= 2 }", - "out[_0, _1] = _in0[_0, 0]*_in1[_0, _1]", - [ - lp.GlobalArg("out", dtype=np.float64, shape=(192, 3)), - lp.GlobalArg("_in0", dtype=np.float64, shape=(192, 1)), - lp.GlobalArg("_in1", dtype=np.float64, shape=(192, 3)), - ], - ) _pt_t_unit_2 = make_loopy_program( "{ [_0, _1] : 0 <= _0 <= 1343 and 0 <= _1 <= 2 }", "out[_0, _1] = in_0[in_1[_0, 0] % 1344, in_2[_0, _1] % 3]", @@ -126,34 +79,14 @@ def _rhs_inner( ) _pt_t_unit_3 = make_loopy_program( "{ [_0, _1] : 0 <= _0 <= 1343 and 0 <= _1 <= 2 }", - "out[_0, _1] = in_0[in_1[_0, 0] % 384, in_2[_0, _1] % 4]", - [ - lp.GlobalArg("out", dtype=np.complex128, shape=(1344, 3)), - lp.GlobalArg("in_0", dtype=np.complex128, shape=(384, 4)), - lp.GlobalArg("in_1", dtype=np.int32, shape=(1344, 1)), - lp.GlobalArg("in_2", dtype=np.int64, shape=(1344, 3)), - ], - ) - _pt_t_unit_4 = make_loopy_program( - "{ [_0, _1] : 0 <= _0 <= 1343 and 0 <= _1 <= 2 }", - "out[_0, _1] = in_0[in_1[_0, 0] % 1344, in_2[_0, _1] % 3]", - [ - lp.GlobalArg("out", dtype=np.complex128, shape=(1344, 3)), - lp.GlobalArg("in_0", dtype=np.complex128, shape=(1344, 3)), - lp.GlobalArg("in_1", dtype=np.int32, shape=(1344, 1)), - lp.GlobalArg("in_2", dtype=np.int64, shape=(1344, 3)), - ], - ) - _pt_t_unit_5 = make_loopy_program( - "{ [_0, _1] : 0 <= _0 <= 1343 and 0 <= _1 <= 2 }", - "out[_0, _1] = cast(np:dtype('complex128'), _in0[_0, 0])*_in1[_0, _1]", + "out[_0, _1] = _in0[_0, 0]*_in1[_0, _1]", [ - lp.GlobalArg("out", dtype=np.complex128, shape=(1344, 3)), + lp.GlobalArg("out", dtype=np.float64, shape=(1344, 3)), lp.GlobalArg("_in0", dtype=np.float64, shape=(1344, 1)), - lp.GlobalArg("_in1", dtype=np.complex128, shape=(1344, 3)), + lp.GlobalArg("_in1", dtype=np.float64, shape=(1344, 3)), ], ) - _pt_t_unit_6 = make_loopy_program( + _pt_t_unit_4 = make_loopy_program( "{ [_0, _1] : 0 <= _0 <= 1535 and 0 <= _1 <= 2 }", "out[_0, _1] = in_0[in_1[_0] % 1, _1]", [ @@ -162,26 +95,26 @@ def _rhs_inner( lp.GlobalArg("in_1", dtype=np.int8, shape=(1536,)), ], ) - _pt_t_unit_7 = make_loopy_program( + _pt_t_unit_5 = make_loopy_program( "{ [_0, _1] : 0 <= _0 <= 1535 and 0 <= _1 <= 2 }", "out[_0, _1] = in_0[in_1[_0, 0] % 1344, in_2[_0, _1] % 3]", [ - lp.GlobalArg("out", dtype=np.complex128, shape=(1536, 3)), - lp.GlobalArg("in_0", dtype=np.complex128, shape=(1344, 3)), + lp.GlobalArg("out", dtype=np.float64, shape=(1536, 3)), + lp.GlobalArg("in_0", dtype=np.float64, shape=(1344, 3)), lp.GlobalArg("in_1", dtype=np.int32, shape=(1536, 1)), lp.GlobalArg("in_2", dtype=np.int64, shape=(1536, 3)), ], ) - _pt_t_unit_8 = make_loopy_program( + _pt_t_unit_6 = make_loopy_program( "{ [_0, _1] : 0 <= _0 <= 1535 and 0 <= _1 <= 2 }", "out[_0, _1] = _in1[_0, _1] if _in0[_0, 0] else 0", [ - lp.GlobalArg("out", dtype=np.complex128, shape=(1536, 3)), + lp.GlobalArg("out", dtype=np.float64, shape=(1536, 3)), lp.GlobalArg("_in0", dtype=np.int8, shape=(1536, 1)), - lp.GlobalArg("_in1", dtype=np.complex128, shape=(1536, 3)), + lp.GlobalArg("_in1", dtype=np.float64, shape=(1536, 3)), ], ) - _pt_t_unit_9 = make_loopy_program( + _pt_t_unit_7 = make_loopy_program( "{ [_0, _1] : 0 <= _0 <= 191 and 0 <= _1 <= 2 }", "out[_0, _1] = in_0[in_1[_0] % 4, _1]", [ @@ -190,6 +123,25 @@ def _rhs_inner( lp.GlobalArg("in_1", dtype=np.int8, shape=(192,)), ], ) + _pt_t_unit_8 = make_loopy_program( + "{ [_0, _1] : 0 <= _0 <= 191 and 0 <= _1 <= 2 }", + "out[_0, _1] = in_0[in_1[_0, 0] % 384, in_2[_0, _1] % 4]", + [ + lp.GlobalArg("out", dtype=np.float64, shape=(192, 3)), + lp.GlobalArg("in_0", dtype=np.float64, shape=(384, 4)), + lp.GlobalArg("in_1", dtype=np.int32, shape=(192, 1)), + lp.GlobalArg("in_2", dtype=np.int64, shape=(192, 3)), + ], + ) + _pt_t_unit_9 = make_loopy_program( + "{ [_0, _1] : 0 <= _0 <= 191 and 0 <= _1 <= 2 }", + "out[_0, _1] = _in0[_0, 0]*_in1[_0, _1]", + [ + lp.GlobalArg("out", dtype=np.float64, shape=(192, 3)), + lp.GlobalArg("_in0", dtype=np.float64, shape=(192, 1)), + lp.GlobalArg("_in1", dtype=np.float64, shape=(192, 3)), + ], + ) _pt_data = actx.thaw(npzfile["_pt_data"]) _pt_tmp_9 = _pt_data[1] _pt_tmp_8 = _pt_tmp_9[:, :, 0] @@ -206,10 +158,10 @@ def _rhs_inner( _pt_tmp_11 = actx.einsum("ij,ikl,jl->jk", _pt_tmp_12, _pt_data_0, _actx_in_1_4_0) del _pt_tmp_13 _pt_tmp_11 = actx.tag((FirstAxisIsElementsTag(),), _pt_tmp_11) - _pt_tmp_10 = (-1 + 0j) * _pt_tmp_11 + _pt_tmp_10 = -1.0 * _pt_tmp_11 _pt_tmp_4 = _pt_tmp_5 + _pt_tmp_10 del _pt_tmp_11 - _pt_tmp_3 = 0j - _pt_tmp_4 + _pt_tmp_3 = 0.0 - _pt_tmp_4 del _pt_tmp_10, _pt_tmp_5 _pt_tmp_2 = -1 * _pt_tmp_3 del _pt_tmp_4 @@ -278,75 +230,75 @@ def _rhs_inner( _actx_in_1_1_0[_pt_tmp_39, _pt_tmp_40] if actx.permits_advanced_indexing else actx.call_loopy( - _pt_t_unit_3, in_0=_actx_in_1_1_0, in_1=_pt_tmp_39, in_2=_pt_tmp_40 + _pt_t_unit_0, in_0=_actx_in_1_1_0, in_1=_pt_tmp_39, in_2=_pt_tmp_40 )["out"] ) _pt_tmp_52 = actx.tag_axis(0, (DiscretizationElementAxisTag(),), _pt_tmp_52) _pt_tmp_52 = actx.tag_axis(1, (DiscretizationDOFAxisTag(),), _pt_tmp_52) - _pt_tmp_51 = 0j + _pt_tmp_52 + _pt_tmp_51 = 0.0 + _pt_tmp_52 _pt_tmp_50 = ( _pt_tmp_51[_pt_tmp_41, _pt_tmp_42] if actx.permits_advanced_indexing else actx.call_loopy( - _pt_t_unit_4, in_0=_pt_tmp_51, in_1=_pt_tmp_41, in_2=_pt_tmp_42 + _pt_t_unit_2, in_0=_pt_tmp_51, in_1=_pt_tmp_41, in_2=_pt_tmp_42 )["out"] ) del _pt_tmp_52 _pt_tmp_50 = actx.tag_axis(0, (DiscretizationElementAxisTag(),), _pt_tmp_50) _pt_tmp_50 = actx.tag_axis(1, (DiscretizationDOFAxisTag(),), _pt_tmp_50) - _pt_tmp_49 = 0j + _pt_tmp_50 + _pt_tmp_49 = 0.0 + _pt_tmp_50 _pt_tmp_48 = _pt_tmp_49 - _pt_tmp_51 del _pt_tmp_50 _pt_tmp_46 = ( _pt_tmp_47 * _pt_tmp_48 if actx.supports_nonscalar_broadcasting - else actx.call_loopy(_pt_t_unit_5, _in0=_pt_tmp_47, _in1=_pt_tmp_48)["out"] + else actx.call_loopy(_pt_t_unit_3, _in0=_pt_tmp_47, _in1=_pt_tmp_48)["out"] ) del _pt_tmp_49, _pt_tmp_51 - _pt_tmp_45 = 0j + _pt_tmp_46 + _pt_tmp_45 = 0.0 + _pt_tmp_46 _pt_tmp_54 = -1.0 * _pt_data_6 del _pt_tmp_46 _pt_tmp_59 = ( _actx_in_1_0_0[_pt_tmp_39, _pt_tmp_40] if actx.permits_advanced_indexing else actx.call_loopy( - _pt_t_unit_3, in_0=_actx_in_1_0_0, in_1=_pt_tmp_39, in_2=_pt_tmp_40 + _pt_t_unit_0, in_0=_actx_in_1_0_0, in_1=_pt_tmp_39, in_2=_pt_tmp_40 )["out"] ) _pt_tmp_59 = actx.tag_axis(0, (DiscretizationElementAxisTag(),), _pt_tmp_59) _pt_tmp_59 = actx.tag_axis(1, (DiscretizationDOFAxisTag(),), _pt_tmp_59) - _pt_tmp_58 = 0j + _pt_tmp_59 + _pt_tmp_58 = 0.0 + _pt_tmp_59 _pt_tmp_57 = ( _pt_tmp_58[_pt_tmp_41, _pt_tmp_42] if actx.permits_advanced_indexing else actx.call_loopy( - _pt_t_unit_4, in_0=_pt_tmp_58, in_1=_pt_tmp_41, in_2=_pt_tmp_42 + _pt_t_unit_2, in_0=_pt_tmp_58, in_1=_pt_tmp_41, in_2=_pt_tmp_42 )["out"] ) del _pt_tmp_59 _pt_tmp_57 = actx.tag_axis(0, (DiscretizationElementAxisTag(),), _pt_tmp_57) _pt_tmp_57 = actx.tag_axis(1, (DiscretizationDOFAxisTag(),), _pt_tmp_57) - _pt_tmp_56 = 0j + _pt_tmp_57 + _pt_tmp_56 = 0.0 + _pt_tmp_57 _pt_tmp_55 = _pt_tmp_56 - _pt_tmp_58 del _pt_tmp_57 _pt_tmp_53 = ( _pt_tmp_54 * _pt_tmp_55 if actx.supports_nonscalar_broadcasting - else actx.call_loopy(_pt_t_unit_5, _in0=_pt_tmp_54, _in1=_pt_tmp_55)["out"] + else actx.call_loopy(_pt_t_unit_3, _in0=_pt_tmp_54, _in1=_pt_tmp_55)["out"] ) del _pt_tmp_56, _pt_tmp_58 _pt_tmp_44 = _pt_tmp_45 + _pt_tmp_53 - _pt_tmp_43 = (0.5 + 0j) * _pt_tmp_44 + _pt_tmp_43 = 0.5 * _pt_tmp_44 del _pt_tmp_45, _pt_tmp_53 _pt_tmp_32 = _pt_tmp_33 - _pt_tmp_43 del _pt_tmp_44 _pt_tmp_30 = ( _pt_tmp_31 * _pt_tmp_32 if actx.supports_nonscalar_broadcasting - else actx.call_loopy(_pt_t_unit_5, _in0=_pt_tmp_31, _in1=_pt_tmp_32)["out"] + else actx.call_loopy(_pt_t_unit_3, _in0=_pt_tmp_31, _in1=_pt_tmp_32)["out"] ) del _pt_tmp_33, _pt_tmp_43 - _pt_tmp_29 = 0j + _pt_tmp_30 + _pt_tmp_29 = 0.0 + _pt_tmp_30 _pt_data_14 = actx.thaw(npzfile["_pt_data_14"]) del _pt_tmp_30 _pt_tmp_61 = -1.0 * _pt_data_14 @@ -354,78 +306,78 @@ def _rhs_inner( _actx_in_1_4_0[_pt_tmp_39, _pt_tmp_40] if actx.permits_advanced_indexing else actx.call_loopy( - _pt_t_unit_3, in_0=_actx_in_1_4_0, in_1=_pt_tmp_39, in_2=_pt_tmp_40 + _pt_t_unit_0, in_0=_actx_in_1_4_0, in_1=_pt_tmp_39, in_2=_pt_tmp_40 )["out"] ) _pt_tmp_68 = actx.tag_axis(0, (DiscretizationElementAxisTag(),), _pt_tmp_68) _pt_tmp_68 = actx.tag_axis(1, (DiscretizationDOFAxisTag(),), _pt_tmp_68) - _pt_tmp_67 = 0j + _pt_tmp_68 + _pt_tmp_67 = 0.0 + _pt_tmp_68 _pt_tmp_66 = ( _pt_tmp_67[_pt_tmp_41, _pt_tmp_42] if actx.permits_advanced_indexing else actx.call_loopy( - _pt_t_unit_4, in_0=_pt_tmp_67, in_1=_pt_tmp_41, in_2=_pt_tmp_42 + _pt_t_unit_2, in_0=_pt_tmp_67, in_1=_pt_tmp_41, in_2=_pt_tmp_42 )["out"] ) del _pt_tmp_68 _pt_tmp_66 = actx.tag_axis(0, (DiscretizationElementAxisTag(),), _pt_tmp_66) _pt_tmp_66 = actx.tag_axis(1, (DiscretizationDOFAxisTag(),), _pt_tmp_66) - _pt_tmp_65 = 0j + _pt_tmp_66 + _pt_tmp_65 = 0.0 + _pt_tmp_66 _pt_tmp_64 = _pt_tmp_65 - _pt_tmp_67 del _pt_tmp_66 - _pt_tmp_63 = (1 + 0j) * _pt_tmp_64 + _pt_tmp_63 = 1.0 * _pt_tmp_64 del _pt_tmp_65, _pt_tmp_67 _pt_tmp_73 = -1.0 * _pt_data_13 _pt_tmp_78 = ( _actx_in_1_2_0[_pt_tmp_39, _pt_tmp_40] if actx.permits_advanced_indexing else actx.call_loopy( - _pt_t_unit_3, in_0=_actx_in_1_2_0, in_1=_pt_tmp_39, in_2=_pt_tmp_40 + _pt_t_unit_0, in_0=_actx_in_1_2_0, in_1=_pt_tmp_39, in_2=_pt_tmp_40 )["out"] ) _pt_tmp_78 = actx.tag_axis(0, (DiscretizationElementAxisTag(),), _pt_tmp_78) _pt_tmp_78 = actx.tag_axis(1, (DiscretizationDOFAxisTag(),), _pt_tmp_78) - _pt_tmp_77 = 0j + _pt_tmp_78 + _pt_tmp_77 = 0.0 + _pt_tmp_78 _pt_tmp_76 = ( _pt_tmp_77[_pt_tmp_41, _pt_tmp_42] if actx.permits_advanced_indexing else actx.call_loopy( - _pt_t_unit_4, in_0=_pt_tmp_77, in_1=_pt_tmp_41, in_2=_pt_tmp_42 + _pt_t_unit_2, in_0=_pt_tmp_77, in_1=_pt_tmp_41, in_2=_pt_tmp_42 )["out"] ) del _pt_tmp_78 _pt_tmp_76 = actx.tag_axis(0, (DiscretizationElementAxisTag(),), _pt_tmp_76) _pt_tmp_76 = actx.tag_axis(1, (DiscretizationDOFAxisTag(),), _pt_tmp_76) - _pt_tmp_75 = 0j + _pt_tmp_76 + _pt_tmp_75 = 0.0 + _pt_tmp_76 _pt_tmp_74 = _pt_tmp_75 - _pt_tmp_77 del _pt_tmp_76 _pt_tmp_72 = ( _pt_tmp_73 * _pt_tmp_74 if actx.supports_nonscalar_broadcasting - else actx.call_loopy(_pt_t_unit_5, _in0=_pt_tmp_73, _in1=_pt_tmp_74)["out"] + else actx.call_loopy(_pt_t_unit_3, _in0=_pt_tmp_73, _in1=_pt_tmp_74)["out"] ) del _pt_tmp_75, _pt_tmp_77 - _pt_tmp_71 = 0j + _pt_tmp_72 + _pt_tmp_71 = 0.0 + _pt_tmp_72 _pt_tmp_80 = 1.0 * _pt_data_14 del _pt_tmp_72 _pt_tmp_79 = ( _pt_tmp_80 * _pt_tmp_55 if actx.supports_nonscalar_broadcasting - else actx.call_loopy(_pt_t_unit_5, _in0=_pt_tmp_80, _in1=_pt_tmp_55)["out"] + else actx.call_loopy(_pt_t_unit_3, _in0=_pt_tmp_80, _in1=_pt_tmp_55)["out"] ) _pt_tmp_70 = _pt_tmp_71 + _pt_tmp_79 - _pt_tmp_69 = (0.5 + 0j) * _pt_tmp_70 + _pt_tmp_69 = 0.5 * _pt_tmp_70 del _pt_tmp_71, _pt_tmp_79 _pt_tmp_62 = _pt_tmp_63 - _pt_tmp_69 del _pt_tmp_70 _pt_tmp_60 = ( _pt_tmp_61 * _pt_tmp_62 if actx.supports_nonscalar_broadcasting - else actx.call_loopy(_pt_t_unit_5, _in0=_pt_tmp_61, _in1=_pt_tmp_62)["out"] + else actx.call_loopy(_pt_t_unit_3, _in0=_pt_tmp_61, _in1=_pt_tmp_62)["out"] ) del _pt_tmp_63, _pt_tmp_69 _pt_tmp_28 = _pt_tmp_29 + _pt_tmp_60 - _pt_tmp_27 = (-0.5 + 0j) * _pt_tmp_28 + _pt_tmp_27 = -0.5 * _pt_tmp_28 del _pt_tmp_29, _pt_tmp_60 _pt_data_15 = actx.thaw(npzfile["_pt_data_15"]) del _pt_tmp_28 @@ -435,26 +387,26 @@ def _rhs_inner( _pt_tmp_82 = ( _pt_data_16[_pt_data_17] if actx.permits_advanced_indexing - else actx.call_loopy(_pt_t_unit_6, in_0=_pt_data_16, in_1=_pt_data_17)["out"] + else actx.call_loopy(_pt_t_unit_4, in_0=_pt_data_16, in_1=_pt_data_17)["out"] ) _pt_tmp_26 = ( _pt_tmp_27[_pt_tmp_81, _pt_tmp_82] if actx.permits_advanced_indexing else actx.call_loopy( - _pt_t_unit_7, in_0=_pt_tmp_27, in_1=_pt_tmp_81, in_2=_pt_tmp_82 + _pt_t_unit_5, in_0=_pt_tmp_27, in_1=_pt_tmp_81, in_2=_pt_tmp_82 )["out"] ) _pt_tmp_24 = ( actx.np.where(_pt_tmp_25, _pt_tmp_26, 0) if actx.supports_nonscalar_broadcasting - else actx.call_loopy(_pt_t_unit_8, _in0=_pt_tmp_25, _in1=_pt_tmp_26)["out"] + else actx.call_loopy(_pt_t_unit_6, _in0=_pt_tmp_25, _in1=_pt_tmp_26)["out"] ) del _pt_tmp_27 _pt_tmp_24 = actx.tag_axis(0, (DiscretizationElementAxisTag(),), _pt_tmp_24) del _pt_tmp_26 _pt_tmp_24 = actx.tag_axis(1, (DiscretizationDOFAxisTag(),), _pt_tmp_24) - _pt_tmp_23 = 0j + _pt_tmp_24 - _pt_tmp_22 = 0j + _pt_tmp_23 + _pt_tmp_23 = 0.0 + _pt_tmp_24 + _pt_tmp_22 = 0.0 + _pt_tmp_23 del _pt_tmp_24 _pt_data_18 = actx.thaw(npzfile["_pt_data_18"]) del _pt_tmp_23 @@ -467,13 +419,13 @@ def _rhs_inner( _pt_tmp_102 = ( _pt_data_8[_pt_data_21] if actx.permits_advanced_indexing - else actx.call_loopy(_pt_t_unit_9, in_0=_pt_data_8, in_1=_pt_data_21)["out"] + else actx.call_loopy(_pt_t_unit_7, in_0=_pt_data_8, in_1=_pt_data_21)["out"] ) _pt_tmp_100 = ( _actx_in_1_5_0[_pt_tmp_101, _pt_tmp_102] if actx.permits_advanced_indexing else actx.call_loopy( - _pt_t_unit_10, in_0=_actx_in_1_5_0, in_1=_pt_tmp_101, in_2=_pt_tmp_102 + _pt_t_unit_8, in_0=_actx_in_1_5_0, in_1=_pt_tmp_101, in_2=_pt_tmp_102 )["out"] ) _pt_tmp_100 = actx.tag_axis(0, (DiscretizationElementAxisTag(),), _pt_tmp_100) @@ -489,59 +441,55 @@ def _rhs_inner( _actx_in_1_1_0[_pt_tmp_101, _pt_tmp_102] if actx.permits_advanced_indexing else actx.call_loopy( - _pt_t_unit_11, in_0=_actx_in_1_1_0, in_1=_pt_tmp_101, in_2=_pt_tmp_102 + _pt_t_unit_8, in_0=_actx_in_1_1_0, in_1=_pt_tmp_101, in_2=_pt_tmp_102 )["out"] ) _pt_tmp_111 = actx.tag_axis(0, (DiscretizationElementAxisTag(),), _pt_tmp_111) _pt_tmp_111 = actx.tag_axis(1, (DiscretizationDOFAxisTag(),), _pt_tmp_111) - _pt_tmp_110 = 0j + _pt_tmp_111 + _pt_tmp_110 = 0.0 + _pt_tmp_111 _pt_tmp_109 = -1 * _pt_tmp_110 del _pt_tmp_111 _pt_tmp_108 = _pt_tmp_109 - _pt_tmp_110 _pt_tmp_106 = ( _pt_tmp_107 * _pt_tmp_108 if actx.supports_nonscalar_broadcasting - else actx.call_loopy(_pt_t_unit_12, _in0=_pt_tmp_107, _in1=_pt_tmp_108)[ - "out" - ] + else actx.call_loopy(_pt_t_unit_9, _in0=_pt_tmp_107, _in1=_pt_tmp_108)["out"] ) del _pt_tmp_109, _pt_tmp_110 - _pt_tmp_105 = 0j + _pt_tmp_106 + _pt_tmp_105 = 0.0 + _pt_tmp_106 _pt_tmp_113 = -1.0 * _pt_data_19 del _pt_tmp_106 _pt_tmp_117 = ( _actx_in_1_0_0[_pt_tmp_101, _pt_tmp_102] if actx.permits_advanced_indexing else actx.call_loopy( - _pt_t_unit_11, in_0=_actx_in_1_0_0, in_1=_pt_tmp_101, in_2=_pt_tmp_102 + _pt_t_unit_8, in_0=_actx_in_1_0_0, in_1=_pt_tmp_101, in_2=_pt_tmp_102 )["out"] ) _pt_tmp_117 = actx.tag_axis(0, (DiscretizationElementAxisTag(),), _pt_tmp_117) _pt_tmp_117 = actx.tag_axis(1, (DiscretizationDOFAxisTag(),), _pt_tmp_117) - _pt_tmp_116 = 0j + _pt_tmp_117 + _pt_tmp_116 = 0.0 + _pt_tmp_117 _pt_tmp_115 = -1 * _pt_tmp_116 del _pt_tmp_117 _pt_tmp_114 = _pt_tmp_115 - _pt_tmp_116 _pt_tmp_112 = ( _pt_tmp_113 * _pt_tmp_114 if actx.supports_nonscalar_broadcasting - else actx.call_loopy(_pt_t_unit_12, _in0=_pt_tmp_113, _in1=_pt_tmp_114)[ - "out" - ] + else actx.call_loopy(_pt_t_unit_9, _in0=_pt_tmp_113, _in1=_pt_tmp_114)["out"] ) del _pt_tmp_115, _pt_tmp_116 _pt_tmp_104 = _pt_tmp_105 + _pt_tmp_112 - _pt_tmp_103 = (0.5 + 0j) * _pt_tmp_104 + _pt_tmp_103 = 0.5 * _pt_tmp_104 del _pt_tmp_105, _pt_tmp_112 _pt_tmp_96 = _pt_tmp_97 - _pt_tmp_103 del _pt_tmp_104 _pt_tmp_94 = ( _pt_tmp_95 * _pt_tmp_96 if actx.supports_nonscalar_broadcasting - else actx.call_loopy(_pt_t_unit_12, _in0=_pt_tmp_95, _in1=_pt_tmp_96)["out"] + else actx.call_loopy(_pt_t_unit_9, _in0=_pt_tmp_95, _in1=_pt_tmp_96)["out"] ) del _pt_tmp_103, _pt_tmp_97 - _pt_tmp_93 = 0j + _pt_tmp_94 + _pt_tmp_93 = 0.0 + _pt_tmp_94 _pt_data_23 = actx.thaw(npzfile["_pt_data_23"]) del _pt_tmp_94 _pt_tmp_119 = -1.0 * _pt_data_23 @@ -549,63 +497,57 @@ def _rhs_inner( _actx_in_1_4_0[_pt_tmp_101, _pt_tmp_102] if actx.permits_advanced_indexing else actx.call_loopy( - _pt_t_unit_11, in_0=_actx_in_1_4_0, in_1=_pt_tmp_101, in_2=_pt_tmp_102 + _pt_t_unit_8, in_0=_actx_in_1_4_0, in_1=_pt_tmp_101, in_2=_pt_tmp_102 )["out"] ) _pt_tmp_124 = actx.tag_axis(0, (DiscretizationElementAxisTag(),), _pt_tmp_124) _pt_tmp_124 = actx.tag_axis(1, (DiscretizationDOFAxisTag(),), _pt_tmp_124) - _pt_tmp_123 = 0j + _pt_tmp_124 + _pt_tmp_123 = 0.0 + _pt_tmp_124 _pt_tmp_122 = _pt_tmp_123 - _pt_tmp_123 del _pt_tmp_124 - _pt_tmp_121 = (1 + 0j) * _pt_tmp_122 + _pt_tmp_121 = 1.0 * _pt_tmp_122 del _pt_tmp_123 _pt_tmp_129 = -1.0 * _pt_data_22 _pt_tmp_133 = ( _actx_in_1_2_0[_pt_tmp_101, _pt_tmp_102] if actx.permits_advanced_indexing else actx.call_loopy( - _pt_t_unit_11, in_0=_actx_in_1_2_0, in_1=_pt_tmp_101, in_2=_pt_tmp_102 + _pt_t_unit_8, in_0=_actx_in_1_2_0, in_1=_pt_tmp_101, in_2=_pt_tmp_102 )["out"] ) _pt_tmp_133 = actx.tag_axis(0, (DiscretizationElementAxisTag(),), _pt_tmp_133) _pt_tmp_133 = actx.tag_axis(1, (DiscretizationDOFAxisTag(),), _pt_tmp_133) - _pt_tmp_132 = 0j + _pt_tmp_133 + _pt_tmp_132 = 0.0 + _pt_tmp_133 _pt_tmp_131 = -1 * _pt_tmp_132 del _pt_tmp_133 _pt_tmp_130 = _pt_tmp_131 - _pt_tmp_132 _pt_tmp_128 = ( _pt_tmp_129 * _pt_tmp_130 if actx.supports_nonscalar_broadcasting - else actx.call_loopy(_pt_t_unit_12, _in0=_pt_tmp_129, _in1=_pt_tmp_130)[ - "out" - ] + else actx.call_loopy(_pt_t_unit_9, _in0=_pt_tmp_129, _in1=_pt_tmp_130)["out"] ) del _pt_tmp_131, _pt_tmp_132 - _pt_tmp_127 = 0j + _pt_tmp_128 + _pt_tmp_127 = 0.0 + _pt_tmp_128 _pt_tmp_135 = 1.0 * _pt_data_23 del _pt_tmp_128 _pt_tmp_134 = ( _pt_tmp_135 * _pt_tmp_114 if actx.supports_nonscalar_broadcasting - else actx.call_loopy(_pt_t_unit_12, _in0=_pt_tmp_135, _in1=_pt_tmp_114)[ - "out" - ] + else actx.call_loopy(_pt_t_unit_9, _in0=_pt_tmp_135, _in1=_pt_tmp_114)["out"] ) _pt_tmp_126 = _pt_tmp_127 + _pt_tmp_134 - _pt_tmp_125 = (0.5 + 0j) * _pt_tmp_126 + _pt_tmp_125 = 0.5 * _pt_tmp_126 del _pt_tmp_127, _pt_tmp_134 _pt_tmp_120 = _pt_tmp_121 - _pt_tmp_125 del _pt_tmp_126 _pt_tmp_118 = ( _pt_tmp_119 * _pt_tmp_120 if actx.supports_nonscalar_broadcasting - else actx.call_loopy(_pt_t_unit_12, _in0=_pt_tmp_119, _in1=_pt_tmp_120)[ - "out" - ] + else actx.call_loopy(_pt_t_unit_9, _in0=_pt_tmp_119, _in1=_pt_tmp_120)["out"] ) del _pt_tmp_121, _pt_tmp_125 _pt_tmp_92 = _pt_tmp_93 + _pt_tmp_118 - _pt_tmp_91 = (-0.5 + 0j) * _pt_tmp_92 + _pt_tmp_91 = -0.5 * _pt_tmp_92 del _pt_tmp_118, _pt_tmp_93 _pt_data_24 = actx.thaw(npzfile["_pt_data_24"]) del _pt_tmp_92 @@ -614,22 +556,22 @@ def _rhs_inner( _pt_tmp_91[_pt_tmp_136, _pt_tmp_82] if actx.permits_advanced_indexing else actx.call_loopy( - _pt_t_unit_13, in_0=_pt_tmp_91, in_1=_pt_tmp_136, in_2=_pt_tmp_82 + _pt_t_unit_10, in_0=_pt_tmp_91, in_1=_pt_tmp_136, in_2=_pt_tmp_82 )["out"] ) _pt_tmp_88 = ( actx.np.where(_pt_tmp_89, _pt_tmp_90, 0) if actx.supports_nonscalar_broadcasting - else actx.call_loopy(_pt_t_unit_8, _in0=_pt_tmp_89, _in1=_pt_tmp_90)["out"] + else actx.call_loopy(_pt_t_unit_6, _in0=_pt_tmp_89, _in1=_pt_tmp_90)["out"] ) del _pt_tmp_91 _pt_tmp_88 = actx.tag_axis(0, (DiscretizationElementAxisTag(),), _pt_tmp_88) del _pt_tmp_90 _pt_tmp_88 = actx.tag_axis(1, (DiscretizationDOFAxisTag(),), _pt_tmp_88) - _pt_tmp_87 = 0j + _pt_tmp_88 - _pt_tmp_86 = 0j + _pt_tmp_87 + _pt_tmp_87 = 0.0 + _pt_tmp_88 + _pt_tmp_86 = 0.0 + _pt_tmp_87 del _pt_tmp_88 - _pt_tmp_137 = actx.np.zeros((1536, 3), dtype=np.complex128) + _pt_tmp_137 = actx.np.zeros((1536, 3), dtype=np.float64) del _pt_tmp_87 _pt_tmp_85 = _pt_tmp_86 + _pt_tmp_137 _pt_tmp_84 = _pt_tmp_85 + _pt_tmp_137 @@ -649,7 +591,7 @@ def _rhs_inner( _pt_tmp_14 = actx.tag((FirstAxisIsElementsTag(),), _pt_tmp_14) del _pt_tmp_17 _pt_tmp_1 = _pt_tmp_2 - _pt_tmp_14 - _pt_tmp_0 = _pt_tmp_1 / np.complex128(1 + 0j) + _pt_tmp_0 = _pt_tmp_1 / np.float64(1.0) del _pt_tmp_14, _pt_tmp_2 _pt_tmp_147 = _pt_data[0] del _pt_tmp_1 @@ -667,168 +609,164 @@ def _rhs_inner( ) del _pt_tmp_144 _pt_tmp_149 = actx.tag((FirstAxisIsElementsTag(),), _pt_tmp_149) - _pt_tmp_148 = (1 + 0j) * _pt_tmp_149 + _pt_tmp_148 = 1.0 * _pt_tmp_149 _pt_tmp_142 = _pt_tmp_143 + _pt_tmp_148 del _pt_tmp_149 - _pt_tmp_141 = 0j - _pt_tmp_142 + _pt_tmp_141 = 0.0 - _pt_tmp_142 del _pt_tmp_143, _pt_tmp_148 _pt_tmp_140 = -1 * _pt_tmp_141 del _pt_tmp_142 _pt_tmp_161 = ( _pt_tmp_73 * _pt_tmp_32 if actx.supports_nonscalar_broadcasting - else actx.call_loopy(_pt_t_unit_5, _in0=_pt_tmp_73, _in1=_pt_tmp_32)["out"] + else actx.call_loopy(_pt_t_unit_3, _in0=_pt_tmp_73, _in1=_pt_tmp_32)["out"] ) del _pt_tmp_141 - _pt_tmp_160 = 0j + _pt_tmp_161 + _pt_tmp_160 = 0.0 + _pt_tmp_161 del _pt_tmp_32 _pt_tmp_169 = ( _actx_in_1_3_0[_pt_tmp_39, _pt_tmp_40] if actx.permits_advanced_indexing else actx.call_loopy( - _pt_t_unit_3, in_0=_actx_in_1_3_0, in_1=_pt_tmp_39, in_2=_pt_tmp_40 + _pt_t_unit_0, in_0=_actx_in_1_3_0, in_1=_pt_tmp_39, in_2=_pt_tmp_40 )["out"] ) del _pt_tmp_161 _pt_tmp_169 = actx.tag_axis(0, (DiscretizationElementAxisTag(),), _pt_tmp_169) del _pt_tmp_39, _pt_tmp_40 _pt_tmp_169 = actx.tag_axis(1, (DiscretizationDOFAxisTag(),), _pt_tmp_169) - _pt_tmp_168 = 0j + _pt_tmp_169 + _pt_tmp_168 = 0.0 + _pt_tmp_169 _pt_tmp_167 = ( _pt_tmp_168[_pt_tmp_41, _pt_tmp_42] if actx.permits_advanced_indexing else actx.call_loopy( - _pt_t_unit_4, in_0=_pt_tmp_168, in_1=_pt_tmp_41, in_2=_pt_tmp_42 + _pt_t_unit_2, in_0=_pt_tmp_168, in_1=_pt_tmp_41, in_2=_pt_tmp_42 )["out"] ) del _pt_tmp_169 _pt_tmp_167 = actx.tag_axis(0, (DiscretizationElementAxisTag(),), _pt_tmp_167) del _pt_tmp_41, _pt_tmp_42 _pt_tmp_167 = actx.tag_axis(1, (DiscretizationDOFAxisTag(),), _pt_tmp_167) - _pt_tmp_166 = 0j + _pt_tmp_167 + _pt_tmp_166 = 0.0 + _pt_tmp_167 _pt_tmp_165 = _pt_tmp_166 - _pt_tmp_168 del _pt_tmp_167 - _pt_tmp_164 = (1 + 0j) * _pt_tmp_165 + _pt_tmp_164 = 1.0 * _pt_tmp_165 del _pt_tmp_166, _pt_tmp_168 _pt_tmp_173 = ( _pt_tmp_31 * _pt_tmp_74 if actx.supports_nonscalar_broadcasting - else actx.call_loopy(_pt_t_unit_5, _in0=_pt_tmp_31, _in1=_pt_tmp_74)["out"] + else actx.call_loopy(_pt_t_unit_3, _in0=_pt_tmp_31, _in1=_pt_tmp_74)["out"] ) - _pt_tmp_172 = 0j + _pt_tmp_173 + _pt_tmp_172 = 0.0 + _pt_tmp_173 _pt_tmp_174 = ( _pt_tmp_61 * _pt_tmp_48 if actx.supports_nonscalar_broadcasting - else actx.call_loopy(_pt_t_unit_5, _in0=_pt_tmp_61, _in1=_pt_tmp_48)["out"] + else actx.call_loopy(_pt_t_unit_3, _in0=_pt_tmp_61, _in1=_pt_tmp_48)["out"] ) del _pt_tmp_173 _pt_tmp_171 = _pt_tmp_172 + _pt_tmp_174 - _pt_tmp_170 = (0.5 + 0j) * _pt_tmp_171 + _pt_tmp_170 = 0.5 * _pt_tmp_171 del _pt_tmp_172, _pt_tmp_174 _pt_tmp_163 = _pt_tmp_164 - _pt_tmp_170 del _pt_tmp_171 _pt_tmp_162 = ( _pt_tmp_80 * _pt_tmp_163 if actx.supports_nonscalar_broadcasting - else actx.call_loopy(_pt_t_unit_5, _in0=_pt_tmp_80, _in1=_pt_tmp_163)["out"] + else actx.call_loopy(_pt_t_unit_3, _in0=_pt_tmp_80, _in1=_pt_tmp_163)["out"] ) del _pt_tmp_164, _pt_tmp_170 _pt_tmp_159 = _pt_tmp_160 + _pt_tmp_162 - _pt_tmp_158 = (-0.5 + 0j) * _pt_tmp_159 + _pt_tmp_158 = -0.5 * _pt_tmp_159 del _pt_tmp_160, _pt_tmp_162 _pt_tmp_157 = ( _pt_tmp_158[_pt_tmp_81, _pt_tmp_82] if actx.permits_advanced_indexing else actx.call_loopy( - _pt_t_unit_7, in_0=_pt_tmp_158, in_1=_pt_tmp_81, in_2=_pt_tmp_82 + _pt_t_unit_5, in_0=_pt_tmp_158, in_1=_pt_tmp_81, in_2=_pt_tmp_82 )["out"] ) del _pt_tmp_159 _pt_tmp_156 = ( actx.np.where(_pt_tmp_25, _pt_tmp_157, 0) if actx.supports_nonscalar_broadcasting - else actx.call_loopy(_pt_t_unit_8, _in0=_pt_tmp_25, _in1=_pt_tmp_157)["out"] + else actx.call_loopy(_pt_t_unit_6, _in0=_pt_tmp_25, _in1=_pt_tmp_157)["out"] ) del _pt_tmp_158 _pt_tmp_156 = actx.tag_axis(0, (DiscretizationElementAxisTag(),), _pt_tmp_156) del _pt_tmp_157 _pt_tmp_156 = actx.tag_axis(1, (DiscretizationDOFAxisTag(),), _pt_tmp_156) - _pt_tmp_155 = 0j + _pt_tmp_156 - _pt_tmp_154 = 0j + _pt_tmp_155 + _pt_tmp_155 = 0.0 + _pt_tmp_156 + _pt_tmp_154 = 0.0 + _pt_tmp_155 del _pt_tmp_156 _pt_tmp_185 = ( _pt_tmp_129 * _pt_tmp_96 if actx.supports_nonscalar_broadcasting - else actx.call_loopy(_pt_t_unit_12, _in0=_pt_tmp_129, _in1=_pt_tmp_96)["out"] + else actx.call_loopy(_pt_t_unit_9, _in0=_pt_tmp_129, _in1=_pt_tmp_96)["out"] ) del _pt_tmp_155 - _pt_tmp_184 = 0j + _pt_tmp_185 + _pt_tmp_184 = 0.0 + _pt_tmp_185 del _pt_tmp_96 _pt_tmp_191 = ( _actx_in_1_3_0[_pt_tmp_101, _pt_tmp_102] if actx.permits_advanced_indexing else actx.call_loopy( - _pt_t_unit_11, in_0=_actx_in_1_3_0, in_1=_pt_tmp_101, in_2=_pt_tmp_102 + _pt_t_unit_8, in_0=_actx_in_1_3_0, in_1=_pt_tmp_101, in_2=_pt_tmp_102 )["out"] ) del _pt_tmp_185 _pt_tmp_191 = actx.tag_axis(0, (DiscretizationElementAxisTag(),), _pt_tmp_191) del _pt_tmp_101, _pt_tmp_102 _pt_tmp_191 = actx.tag_axis(1, (DiscretizationDOFAxisTag(),), _pt_tmp_191) - _pt_tmp_190 = 0j + _pt_tmp_191 + _pt_tmp_190 = 0.0 + _pt_tmp_191 _pt_tmp_189 = _pt_tmp_190 - _pt_tmp_190 del _pt_tmp_191 - _pt_tmp_188 = (1 + 0j) * _pt_tmp_189 + _pt_tmp_188 = 1.0 * _pt_tmp_189 del _pt_tmp_190 _pt_tmp_195 = ( _pt_tmp_95 * _pt_tmp_130 if actx.supports_nonscalar_broadcasting - else actx.call_loopy(_pt_t_unit_12, _in0=_pt_tmp_95, _in1=_pt_tmp_130)["out"] + else actx.call_loopy(_pt_t_unit_9, _in0=_pt_tmp_95, _in1=_pt_tmp_130)["out"] ) - _pt_tmp_194 = 0j + _pt_tmp_195 + _pt_tmp_194 = 0.0 + _pt_tmp_195 _pt_tmp_196 = ( _pt_tmp_119 * _pt_tmp_108 if actx.supports_nonscalar_broadcasting - else actx.call_loopy(_pt_t_unit_12, _in0=_pt_tmp_119, _in1=_pt_tmp_108)[ - "out" - ] + else actx.call_loopy(_pt_t_unit_9, _in0=_pt_tmp_119, _in1=_pt_tmp_108)["out"] ) del _pt_tmp_195 _pt_tmp_193 = _pt_tmp_194 + _pt_tmp_196 - _pt_tmp_192 = (0.5 + 0j) * _pt_tmp_193 + _pt_tmp_192 = 0.5 * _pt_tmp_193 del _pt_tmp_194, _pt_tmp_196 _pt_tmp_187 = _pt_tmp_188 - _pt_tmp_192 del _pt_tmp_193 _pt_tmp_186 = ( _pt_tmp_135 * _pt_tmp_187 if actx.supports_nonscalar_broadcasting - else actx.call_loopy(_pt_t_unit_12, _in0=_pt_tmp_135, _in1=_pt_tmp_187)[ - "out" - ] + else actx.call_loopy(_pt_t_unit_9, _in0=_pt_tmp_135, _in1=_pt_tmp_187)["out"] ) del _pt_tmp_188, _pt_tmp_192 _pt_tmp_183 = _pt_tmp_184 + _pt_tmp_186 - _pt_tmp_182 = (-0.5 + 0j) * _pt_tmp_183 + _pt_tmp_182 = -0.5 * _pt_tmp_183 del _pt_tmp_184, _pt_tmp_186 _pt_tmp_181 = ( _pt_tmp_182[_pt_tmp_136, _pt_tmp_82] if actx.permits_advanced_indexing else actx.call_loopy( - _pt_t_unit_13, in_0=_pt_tmp_182, in_1=_pt_tmp_136, in_2=_pt_tmp_82 + _pt_t_unit_10, in_0=_pt_tmp_182, in_1=_pt_tmp_136, in_2=_pt_tmp_82 )["out"] ) del _pt_tmp_183 _pt_tmp_180 = ( actx.np.where(_pt_tmp_89, _pt_tmp_181, 0) if actx.supports_nonscalar_broadcasting - else actx.call_loopy(_pt_t_unit_8, _in0=_pt_tmp_89, _in1=_pt_tmp_181)["out"] + else actx.call_loopy(_pt_t_unit_6, _in0=_pt_tmp_89, _in1=_pt_tmp_181)["out"] ) del _pt_tmp_182 _pt_tmp_180 = actx.tag_axis(0, (DiscretizationElementAxisTag(),), _pt_tmp_180) del _pt_tmp_181 _pt_tmp_180 = actx.tag_axis(1, (DiscretizationDOFAxisTag(),), _pt_tmp_180) - _pt_tmp_179 = 0j + _pt_tmp_180 - _pt_tmp_178 = 0j + _pt_tmp_179 + _pt_tmp_179 = 0.0 + _pt_tmp_180 + _pt_tmp_178 = 0.0 + _pt_tmp_179 del _pt_tmp_180 _pt_tmp_177 = _pt_tmp_178 + _pt_tmp_137 del _pt_tmp_179 @@ -849,105 +787,101 @@ def _rhs_inner( _pt_tmp_150 = actx.tag((FirstAxisIsElementsTag(),), _pt_tmp_150) del _pt_tmp_151 _pt_tmp_139 = _pt_tmp_140 - _pt_tmp_150 - _pt_tmp_138 = _pt_tmp_139 / np.complex128(1 + 0j) + _pt_tmp_138 = _pt_tmp_139 / np.float64(1.0) del _pt_tmp_140, _pt_tmp_150 _pt_tmp_204 = actx.einsum( "ij,ikl,jl->jk", _pt_tmp_146, _pt_data_0, _actx_in_1_4_0 ) del _pt_tmp_139 _pt_tmp_204 = actx.tag((FirstAxisIsElementsTag(),), _pt_tmp_204) - _pt_tmp_203 = (1 + 0j) * _pt_tmp_204 - _pt_tmp_202 = 0j + _pt_tmp_203 + _pt_tmp_203 = 1.0 * _pt_tmp_204 + _pt_tmp_202 = 0.0 + _pt_tmp_203 del _pt_tmp_204 _pt_tmp_206 = actx.einsum("ij,ikl,jl->jk", _pt_tmp_8, _pt_data_0, _actx_in_1_3_0) del _pt_tmp_203 _pt_tmp_206 = actx.tag((FirstAxisIsElementsTag(),), _pt_tmp_206) - _pt_tmp_205 = (-1 + 0j) * _pt_tmp_206 + _pt_tmp_205 = -1.0 * _pt_tmp_206 _pt_tmp_201 = _pt_tmp_202 + _pt_tmp_205 del _pt_tmp_206 - _pt_tmp_200 = 0j - _pt_tmp_201 + _pt_tmp_200 = 0.0 - _pt_tmp_201 del _pt_tmp_202, _pt_tmp_205 _pt_tmp_199 = -1 * _pt_tmp_200 del _pt_tmp_201 _pt_tmp_218 = ( _pt_tmp_47 * _pt_tmp_62 if actx.supports_nonscalar_broadcasting - else actx.call_loopy(_pt_t_unit_5, _in0=_pt_tmp_47, _in1=_pt_tmp_62)["out"] + else actx.call_loopy(_pt_t_unit_3, _in0=_pt_tmp_47, _in1=_pt_tmp_62)["out"] ) del _pt_tmp_200 - _pt_tmp_217 = 0j + _pt_tmp_218 + _pt_tmp_217 = 0.0 + _pt_tmp_218 del _pt_tmp_62 _pt_tmp_219 = ( _pt_tmp_54 * _pt_tmp_163 if actx.supports_nonscalar_broadcasting - else actx.call_loopy(_pt_t_unit_5, _in0=_pt_tmp_54, _in1=_pt_tmp_163)["out"] + else actx.call_loopy(_pt_t_unit_3, _in0=_pt_tmp_54, _in1=_pt_tmp_163)["out"] ) del _pt_tmp_218 _pt_tmp_216 = _pt_tmp_217 + _pt_tmp_219 del _pt_tmp_163 - _pt_tmp_215 = (-0.5 + 0j) * _pt_tmp_216 + _pt_tmp_215 = -0.5 * _pt_tmp_216 del _pt_tmp_217, _pt_tmp_219 _pt_tmp_214 = ( _pt_tmp_215[_pt_tmp_81, _pt_tmp_82] if actx.permits_advanced_indexing else actx.call_loopy( - _pt_t_unit_7, in_0=_pt_tmp_215, in_1=_pt_tmp_81, in_2=_pt_tmp_82 + _pt_t_unit_5, in_0=_pt_tmp_215, in_1=_pt_tmp_81, in_2=_pt_tmp_82 )["out"] ) del _pt_tmp_216 _pt_tmp_213 = ( actx.np.where(_pt_tmp_25, _pt_tmp_214, 0) if actx.supports_nonscalar_broadcasting - else actx.call_loopy(_pt_t_unit_8, _in0=_pt_tmp_25, _in1=_pt_tmp_214)["out"] + else actx.call_loopy(_pt_t_unit_6, _in0=_pt_tmp_25, _in1=_pt_tmp_214)["out"] ) del _pt_tmp_215 _pt_tmp_213 = actx.tag_axis(0, (DiscretizationElementAxisTag(),), _pt_tmp_213) del _pt_tmp_214 _pt_tmp_213 = actx.tag_axis(1, (DiscretizationDOFAxisTag(),), _pt_tmp_213) - _pt_tmp_212 = 0j + _pt_tmp_213 - _pt_tmp_211 = 0j + _pt_tmp_212 + _pt_tmp_212 = 0.0 + _pt_tmp_213 + _pt_tmp_211 = 0.0 + _pt_tmp_212 del _pt_tmp_213 _pt_tmp_230 = ( _pt_tmp_107 * _pt_tmp_120 if actx.supports_nonscalar_broadcasting - else actx.call_loopy(_pt_t_unit_12, _in0=_pt_tmp_107, _in1=_pt_tmp_120)[ - "out" - ] + else actx.call_loopy(_pt_t_unit_9, _in0=_pt_tmp_107, _in1=_pt_tmp_120)["out"] ) del _pt_tmp_212 - _pt_tmp_229 = 0j + _pt_tmp_230 + _pt_tmp_229 = 0.0 + _pt_tmp_230 del _pt_tmp_120 _pt_tmp_231 = ( _pt_tmp_113 * _pt_tmp_187 if actx.supports_nonscalar_broadcasting - else actx.call_loopy(_pt_t_unit_12, _in0=_pt_tmp_113, _in1=_pt_tmp_187)[ - "out" - ] + else actx.call_loopy(_pt_t_unit_9, _in0=_pt_tmp_113, _in1=_pt_tmp_187)["out"] ) del _pt_tmp_230 _pt_tmp_228 = _pt_tmp_229 + _pt_tmp_231 del _pt_tmp_187 - _pt_tmp_227 = (-0.5 + 0j) * _pt_tmp_228 + _pt_tmp_227 = -0.5 * _pt_tmp_228 del _pt_tmp_229, _pt_tmp_231 _pt_tmp_226 = ( _pt_tmp_227[_pt_tmp_136, _pt_tmp_82] if actx.permits_advanced_indexing else actx.call_loopy( - _pt_t_unit_13, in_0=_pt_tmp_227, in_1=_pt_tmp_136, in_2=_pt_tmp_82 + _pt_t_unit_10, in_0=_pt_tmp_227, in_1=_pt_tmp_136, in_2=_pt_tmp_82 )["out"] ) del _pt_tmp_228 _pt_tmp_225 = ( actx.np.where(_pt_tmp_89, _pt_tmp_226, 0) if actx.supports_nonscalar_broadcasting - else actx.call_loopy(_pt_t_unit_8, _in0=_pt_tmp_89, _in1=_pt_tmp_226)["out"] + else actx.call_loopy(_pt_t_unit_6, _in0=_pt_tmp_89, _in1=_pt_tmp_226)["out"] ) del _pt_tmp_227 _pt_tmp_225 = actx.tag_axis(0, (DiscretizationElementAxisTag(),), _pt_tmp_225) del _pt_tmp_226 _pt_tmp_225 = actx.tag_axis(1, (DiscretizationDOFAxisTag(),), _pt_tmp_225) - _pt_tmp_224 = 0j + _pt_tmp_225 - _pt_tmp_223 = 0j + _pt_tmp_224 + _pt_tmp_224 = 0.0 + _pt_tmp_225 + _pt_tmp_223 = 0.0 + _pt_tmp_224 del _pt_tmp_225 _pt_tmp_222 = _pt_tmp_223 + _pt_tmp_137 del _pt_tmp_224 @@ -968,185 +902,177 @@ def _rhs_inner( _pt_tmp_207 = actx.tag((FirstAxisIsElementsTag(),), _pt_tmp_207) del _pt_tmp_208 _pt_tmp_198 = _pt_tmp_199 - _pt_tmp_207 - _pt_tmp_197 = _pt_tmp_198 / np.complex128(1 + 0j) + _pt_tmp_197 = _pt_tmp_198 / np.float64(1.0) del _pt_tmp_199, _pt_tmp_207 _pt_tmp_238 = actx.einsum("ij,ikl,jl->jk", _pt_tmp_8, _pt_data_0, _actx_in_1_2_0) del _pt_tmp_198 _pt_tmp_238 = actx.tag((FirstAxisIsElementsTag(),), _pt_tmp_238) - _pt_tmp_237 = (1 + 0j) * _pt_tmp_238 - _pt_tmp_236 = 0j + _pt_tmp_237 + _pt_tmp_237 = 1.0 * _pt_tmp_238 + _pt_tmp_236 = 0.0 + _pt_tmp_237 del _pt_tmp_238 _pt_tmp_240 = actx.einsum( "ij,ikl,jl->jk", _pt_tmp_12, _pt_data_0, _actx_in_1_1_0 ) del _pt_tmp_237 _pt_tmp_240 = actx.tag((FirstAxisIsElementsTag(),), _pt_tmp_240) - _pt_tmp_239 = (-1 + 0j) * _pt_tmp_240 + _pt_tmp_239 = -1.0 * _pt_tmp_240 _pt_tmp_235 = _pt_tmp_236 + _pt_tmp_239 del _pt_tmp_240 _pt_tmp_234 = -1 * _pt_tmp_235 del _pt_tmp_236, _pt_tmp_239 - _pt_tmp_254 = (1 + 0j) * _pt_tmp_74 + _pt_tmp_254 = 1.0 * _pt_tmp_74 del _pt_tmp_235 _pt_tmp_258 = ( _pt_tmp_47 * _pt_tmp_64 if actx.supports_nonscalar_broadcasting - else actx.call_loopy(_pt_t_unit_5, _in0=_pt_tmp_47, _in1=_pt_tmp_64)["out"] + else actx.call_loopy(_pt_t_unit_3, _in0=_pt_tmp_47, _in1=_pt_tmp_64)["out"] ) del _pt_tmp_74 - _pt_tmp_257 = 0j + _pt_tmp_258 + _pt_tmp_257 = 0.0 + _pt_tmp_258 _pt_tmp_259 = ( _pt_tmp_54 * _pt_tmp_165 if actx.supports_nonscalar_broadcasting - else actx.call_loopy(_pt_t_unit_5, _in0=_pt_tmp_54, _in1=_pt_tmp_165)["out"] + else actx.call_loopy(_pt_t_unit_3, _in0=_pt_tmp_54, _in1=_pt_tmp_165)["out"] ) del _pt_tmp_258 _pt_tmp_256 = _pt_tmp_257 + _pt_tmp_259 - _pt_tmp_255 = (0.5 + 0j) * _pt_tmp_256 + _pt_tmp_255 = 0.5 * _pt_tmp_256 del _pt_tmp_257, _pt_tmp_259 _pt_tmp_253 = _pt_tmp_254 + _pt_tmp_255 del _pt_tmp_256 _pt_tmp_252 = ( _pt_tmp_31 * _pt_tmp_253 if actx.supports_nonscalar_broadcasting - else actx.call_loopy(_pt_t_unit_5, _in0=_pt_tmp_31, _in1=_pt_tmp_253)["out"] + else actx.call_loopy(_pt_t_unit_3, _in0=_pt_tmp_31, _in1=_pt_tmp_253)["out"] ) del _pt_tmp_254, _pt_tmp_255 - _pt_tmp_251 = 0j + _pt_tmp_252 - _pt_tmp_262 = (1 + 0j) * _pt_tmp_48 + _pt_tmp_251 = 0.0 + _pt_tmp_252 + _pt_tmp_262 = 1.0 * _pt_tmp_48 del _pt_tmp_252 _pt_tmp_266 = ( _pt_tmp_73 * _pt_tmp_34 if actx.supports_nonscalar_broadcasting - else actx.call_loopy(_pt_t_unit_14, _in0=_pt_tmp_73, _in1=_pt_tmp_34)["out"] + else actx.call_loopy(_pt_t_unit_3, _in0=_pt_tmp_73, _in1=_pt_tmp_34)["out"] ) del _pt_tmp_48 _pt_tmp_265 = 0.0 + _pt_tmp_266 _pt_tmp_267 = ( _pt_tmp_80 * _pt_tmp_165 if actx.supports_nonscalar_broadcasting - else actx.call_loopy(_pt_t_unit_5, _in0=_pt_tmp_80, _in1=_pt_tmp_165)["out"] + else actx.call_loopy(_pt_t_unit_3, _in0=_pt_tmp_80, _in1=_pt_tmp_165)["out"] ) del _pt_tmp_266 _pt_tmp_264 = _pt_tmp_265 + _pt_tmp_267 del _pt_tmp_165 - _pt_tmp_263 = (0.5 + 0j) * _pt_tmp_264 + _pt_tmp_263 = 0.5 * _pt_tmp_264 del _pt_tmp_265, _pt_tmp_267 _pt_tmp_261 = _pt_tmp_262 + _pt_tmp_263 del _pt_tmp_264 _pt_tmp_260 = ( _pt_tmp_61 * _pt_tmp_261 if actx.supports_nonscalar_broadcasting - else actx.call_loopy(_pt_t_unit_5, _in0=_pt_tmp_61, _in1=_pt_tmp_261)["out"] + else actx.call_loopy(_pt_t_unit_3, _in0=_pt_tmp_61, _in1=_pt_tmp_261)["out"] ) del _pt_tmp_262, _pt_tmp_263 _pt_tmp_250 = _pt_tmp_251 + _pt_tmp_260 - _pt_tmp_249 = (0.5 + 0j) * _pt_tmp_250 + _pt_tmp_249 = 0.5 * _pt_tmp_250 del _pt_tmp_251, _pt_tmp_260 _pt_tmp_248 = ( _pt_tmp_249[_pt_tmp_81, _pt_tmp_82] if actx.permits_advanced_indexing else actx.call_loopy( - _pt_t_unit_7, in_0=_pt_tmp_249, in_1=_pt_tmp_81, in_2=_pt_tmp_82 + _pt_t_unit_5, in_0=_pt_tmp_249, in_1=_pt_tmp_81, in_2=_pt_tmp_82 )["out"] ) del _pt_tmp_250 _pt_tmp_247 = ( actx.np.where(_pt_tmp_25, _pt_tmp_248, 0) if actx.supports_nonscalar_broadcasting - else actx.call_loopy(_pt_t_unit_8, _in0=_pt_tmp_25, _in1=_pt_tmp_248)["out"] + else actx.call_loopy(_pt_t_unit_6, _in0=_pt_tmp_25, _in1=_pt_tmp_248)["out"] ) del _pt_tmp_249 _pt_tmp_247 = actx.tag_axis(0, (DiscretizationElementAxisTag(),), _pt_tmp_247) del _pt_tmp_248 _pt_tmp_247 = actx.tag_axis(1, (DiscretizationDOFAxisTag(),), _pt_tmp_247) - _pt_tmp_246 = 0j + _pt_tmp_247 - _pt_tmp_245 = 0j + _pt_tmp_246 + _pt_tmp_246 = 0.0 + _pt_tmp_247 + _pt_tmp_245 = 0.0 + _pt_tmp_246 del _pt_tmp_247 - _pt_tmp_280 = (1 + 0j) * _pt_tmp_130 + _pt_tmp_280 = 1.0 * _pt_tmp_130 del _pt_tmp_246 _pt_tmp_284 = ( _pt_tmp_107 * _pt_tmp_122 if actx.supports_nonscalar_broadcasting - else actx.call_loopy(_pt_t_unit_12, _in0=_pt_tmp_107, _in1=_pt_tmp_122)[ - "out" - ] + else actx.call_loopy(_pt_t_unit_9, _in0=_pt_tmp_107, _in1=_pt_tmp_122)["out"] ) del _pt_tmp_130 - _pt_tmp_283 = 0j + _pt_tmp_284 + _pt_tmp_283 = 0.0 + _pt_tmp_284 _pt_tmp_285 = ( _pt_tmp_113 * _pt_tmp_189 if actx.supports_nonscalar_broadcasting - else actx.call_loopy(_pt_t_unit_12, _in0=_pt_tmp_113, _in1=_pt_tmp_189)[ - "out" - ] + else actx.call_loopy(_pt_t_unit_9, _in0=_pt_tmp_113, _in1=_pt_tmp_189)["out"] ) del _pt_tmp_284 _pt_tmp_282 = _pt_tmp_283 + _pt_tmp_285 - _pt_tmp_281 = (0.5 + 0j) * _pt_tmp_282 + _pt_tmp_281 = 0.5 * _pt_tmp_282 del _pt_tmp_283, _pt_tmp_285 _pt_tmp_279 = _pt_tmp_280 + _pt_tmp_281 del _pt_tmp_282 _pt_tmp_278 = ( _pt_tmp_95 * _pt_tmp_279 if actx.supports_nonscalar_broadcasting - else actx.call_loopy(_pt_t_unit_12, _in0=_pt_tmp_95, _in1=_pt_tmp_279)["out"] + else actx.call_loopy(_pt_t_unit_9, _in0=_pt_tmp_95, _in1=_pt_tmp_279)["out"] ) del _pt_tmp_280, _pt_tmp_281 - _pt_tmp_277 = 0j + _pt_tmp_278 - _pt_tmp_288 = (1 + 0j) * _pt_tmp_108 + _pt_tmp_277 = 0.0 + _pt_tmp_278 + _pt_tmp_288 = 1.0 * _pt_tmp_108 del _pt_tmp_278 _pt_tmp_292 = ( _pt_tmp_129 * _pt_tmp_98 if actx.supports_nonscalar_broadcasting - else actx.call_loopy(_pt_t_unit_15, _in0=_pt_tmp_129, _in1=_pt_tmp_98)["out"] + else actx.call_loopy(_pt_t_unit_9, _in0=_pt_tmp_129, _in1=_pt_tmp_98)["out"] ) del _pt_tmp_108 _pt_tmp_291 = 0.0 + _pt_tmp_292 _pt_tmp_293 = ( _pt_tmp_135 * _pt_tmp_189 if actx.supports_nonscalar_broadcasting - else actx.call_loopy(_pt_t_unit_12, _in0=_pt_tmp_135, _in1=_pt_tmp_189)[ - "out" - ] + else actx.call_loopy(_pt_t_unit_9, _in0=_pt_tmp_135, _in1=_pt_tmp_189)["out"] ) del _pt_tmp_292 _pt_tmp_290 = _pt_tmp_291 + _pt_tmp_293 del _pt_tmp_189 - _pt_tmp_289 = (0.5 + 0j) * _pt_tmp_290 + _pt_tmp_289 = 0.5 * _pt_tmp_290 del _pt_tmp_291, _pt_tmp_293 _pt_tmp_287 = _pt_tmp_288 + _pt_tmp_289 del _pt_tmp_290 _pt_tmp_286 = ( _pt_tmp_119 * _pt_tmp_287 if actx.supports_nonscalar_broadcasting - else actx.call_loopy(_pt_t_unit_12, _in0=_pt_tmp_119, _in1=_pt_tmp_287)[ - "out" - ] + else actx.call_loopy(_pt_t_unit_9, _in0=_pt_tmp_119, _in1=_pt_tmp_287)["out"] ) del _pt_tmp_288, _pt_tmp_289 _pt_tmp_276 = _pt_tmp_277 + _pt_tmp_286 - _pt_tmp_275 = (0.5 + 0j) * _pt_tmp_276 + _pt_tmp_275 = 0.5 * _pt_tmp_276 del _pt_tmp_277, _pt_tmp_286 _pt_tmp_274 = ( _pt_tmp_275[_pt_tmp_136, _pt_tmp_82] if actx.permits_advanced_indexing else actx.call_loopy( - _pt_t_unit_13, in_0=_pt_tmp_275, in_1=_pt_tmp_136, in_2=_pt_tmp_82 + _pt_t_unit_10, in_0=_pt_tmp_275, in_1=_pt_tmp_136, in_2=_pt_tmp_82 )["out"] ) del _pt_tmp_276 _pt_tmp_273 = ( actx.np.where(_pt_tmp_89, _pt_tmp_274, 0) if actx.supports_nonscalar_broadcasting - else actx.call_loopy(_pt_t_unit_8, _in0=_pt_tmp_89, _in1=_pt_tmp_274)["out"] + else actx.call_loopy(_pt_t_unit_6, _in0=_pt_tmp_89, _in1=_pt_tmp_274)["out"] ) del _pt_tmp_275 _pt_tmp_273 = actx.tag_axis(0, (DiscretizationElementAxisTag(),), _pt_tmp_273) del _pt_tmp_274 _pt_tmp_273 = actx.tag_axis(1, (DiscretizationDOFAxisTag(),), _pt_tmp_273) - _pt_tmp_272 = 0j + _pt_tmp_273 - _pt_tmp_271 = 0j + _pt_tmp_272 + _pt_tmp_272 = 0.0 + _pt_tmp_273 + _pt_tmp_271 = 0.0 + _pt_tmp_272 del _pt_tmp_273 _pt_tmp_270 = _pt_tmp_271 + _pt_tmp_137 del _pt_tmp_272 @@ -1167,15 +1093,15 @@ def _rhs_inner( _pt_tmp_241 = actx.tag((FirstAxisIsElementsTag(),), _pt_tmp_241) del _pt_tmp_242 _pt_tmp_233 = _pt_tmp_234 - _pt_tmp_241 - _pt_tmp_232 = _pt_tmp_233 / np.complex128(1 + 0j) + _pt_tmp_232 = _pt_tmp_233 / np.float64(1.0) del _pt_tmp_234, _pt_tmp_241 _pt_tmp_300 = actx.einsum( "ij,ikl,jl->jk", _pt_tmp_146, _pt_data_0, _actx_in_1_2_0 ) del _pt_tmp_233 _pt_tmp_300 = actx.tag((FirstAxisIsElementsTag(),), _pt_tmp_300) - _pt_tmp_299 = (-1 + 0j) * _pt_tmp_300 - _pt_tmp_298 = 0j + _pt_tmp_299 + _pt_tmp_299 = -1.0 * _pt_tmp_300 + _pt_tmp_298 = 0.0 + _pt_tmp_299 del _pt_tmp_300 _pt_tmp_302 = actx.einsum( "ij,ikl,jl->jk", _pt_tmp_12, _pt_data_0, _actx_in_1_0_0 @@ -1183,7 +1109,7 @@ def _rhs_inner( del _pt_tmp_299 _pt_tmp_302 = actx.tag((FirstAxisIsElementsTag(),), _pt_tmp_302) del _pt_tmp_12 - _pt_tmp_301 = (1 + 0j) * _pt_tmp_302 + _pt_tmp_301 = 1.0 * _pt_tmp_302 _pt_tmp_297 = _pt_tmp_298 + _pt_tmp_301 del _pt_tmp_302 _pt_tmp_296 = -1 * _pt_tmp_297 @@ -1191,17 +1117,17 @@ def _rhs_inner( _pt_tmp_314 = ( _pt_tmp_73 * _pt_tmp_253 if actx.supports_nonscalar_broadcasting - else actx.call_loopy(_pt_t_unit_5, _in0=_pt_tmp_73, _in1=_pt_tmp_253)["out"] + else actx.call_loopy(_pt_t_unit_3, _in0=_pt_tmp_73, _in1=_pt_tmp_253)["out"] ) del _pt_tmp_297 - _pt_tmp_313 = 0j + _pt_tmp_314 + _pt_tmp_313 = 0.0 + _pt_tmp_314 del _pt_tmp_253, _pt_tmp_73 - _pt_tmp_317 = (1 + 0j) * _pt_tmp_55 + _pt_tmp_317 = 1.0 * _pt_tmp_55 del _pt_tmp_314 _pt_tmp_321 = ( _pt_tmp_31 * _pt_tmp_34 if actx.supports_nonscalar_broadcasting - else actx.call_loopy(_pt_t_unit_14, _in0=_pt_tmp_31, _in1=_pt_tmp_34)["out"] + else actx.call_loopy(_pt_t_unit_3, _in0=_pt_tmp_31, _in1=_pt_tmp_34)["out"] ) del _pt_tmp_55 _pt_tmp_320 = 0.0 + _pt_tmp_321 @@ -1209,61 +1135,59 @@ def _rhs_inner( _pt_tmp_322 = ( _pt_tmp_61 * _pt_tmp_64 if actx.supports_nonscalar_broadcasting - else actx.call_loopy(_pt_t_unit_5, _in0=_pt_tmp_61, _in1=_pt_tmp_64)["out"] + else actx.call_loopy(_pt_t_unit_3, _in0=_pt_tmp_61, _in1=_pt_tmp_64)["out"] ) del _pt_tmp_321 _pt_tmp_319 = _pt_tmp_320 + _pt_tmp_322 del _pt_tmp_61, _pt_tmp_64 - _pt_tmp_318 = (0.5 + 0j) * _pt_tmp_319 + _pt_tmp_318 = 0.5 * _pt_tmp_319 del _pt_tmp_320, _pt_tmp_322 _pt_tmp_316 = _pt_tmp_317 + _pt_tmp_318 del _pt_tmp_319 _pt_tmp_315 = ( _pt_tmp_80 * _pt_tmp_316 if actx.supports_nonscalar_broadcasting - else actx.call_loopy(_pt_t_unit_5, _in0=_pt_tmp_80, _in1=_pt_tmp_316)["out"] + else actx.call_loopy(_pt_t_unit_3, _in0=_pt_tmp_80, _in1=_pt_tmp_316)["out"] ) del _pt_tmp_317, _pt_tmp_318 _pt_tmp_312 = _pt_tmp_313 + _pt_tmp_315 del _pt_tmp_80 - _pt_tmp_311 = (0.5 + 0j) * _pt_tmp_312 + _pt_tmp_311 = 0.5 * _pt_tmp_312 del _pt_tmp_313, _pt_tmp_315 _pt_tmp_310 = ( _pt_tmp_311[_pt_tmp_81, _pt_tmp_82] if actx.permits_advanced_indexing else actx.call_loopy( - _pt_t_unit_7, in_0=_pt_tmp_311, in_1=_pt_tmp_81, in_2=_pt_tmp_82 + _pt_t_unit_5, in_0=_pt_tmp_311, in_1=_pt_tmp_81, in_2=_pt_tmp_82 )["out"] ) del _pt_tmp_312 _pt_tmp_309 = ( actx.np.where(_pt_tmp_25, _pt_tmp_310, 0) if actx.supports_nonscalar_broadcasting - else actx.call_loopy(_pt_t_unit_8, _in0=_pt_tmp_25, _in1=_pt_tmp_310)["out"] + else actx.call_loopy(_pt_t_unit_6, _in0=_pt_tmp_25, _in1=_pt_tmp_310)["out"] ) del _pt_tmp_311 _pt_tmp_309 = actx.tag_axis(0, (DiscretizationElementAxisTag(),), _pt_tmp_309) del _pt_tmp_310 _pt_tmp_309 = actx.tag_axis(1, (DiscretizationDOFAxisTag(),), _pt_tmp_309) - _pt_tmp_308 = 0j + _pt_tmp_309 - _pt_tmp_307 = 0j + _pt_tmp_308 + _pt_tmp_308 = 0.0 + _pt_tmp_309 + _pt_tmp_307 = 0.0 + _pt_tmp_308 del _pt_tmp_309 _pt_tmp_333 = ( _pt_tmp_129 * _pt_tmp_279 if actx.supports_nonscalar_broadcasting - else actx.call_loopy(_pt_t_unit_12, _in0=_pt_tmp_129, _in1=_pt_tmp_279)[ - "out" - ] + else actx.call_loopy(_pt_t_unit_9, _in0=_pt_tmp_129, _in1=_pt_tmp_279)["out"] ) del _pt_tmp_308 - _pt_tmp_332 = 0j + _pt_tmp_333 + _pt_tmp_332 = 0.0 + _pt_tmp_333 del _pt_tmp_129, _pt_tmp_279 - _pt_tmp_336 = (1 + 0j) * _pt_tmp_114 + _pt_tmp_336 = 1.0 * _pt_tmp_114 del _pt_tmp_333 _pt_tmp_340 = ( _pt_tmp_95 * _pt_tmp_98 if actx.supports_nonscalar_broadcasting - else actx.call_loopy(_pt_t_unit_15, _in0=_pt_tmp_95, _in1=_pt_tmp_98)["out"] + else actx.call_loopy(_pt_t_unit_9, _in0=_pt_tmp_95, _in1=_pt_tmp_98)["out"] ) del _pt_tmp_114 _pt_tmp_339 = 0.0 + _pt_tmp_340 @@ -1271,48 +1195,44 @@ def _rhs_inner( _pt_tmp_341 = ( _pt_tmp_119 * _pt_tmp_122 if actx.supports_nonscalar_broadcasting - else actx.call_loopy(_pt_t_unit_12, _in0=_pt_tmp_119, _in1=_pt_tmp_122)[ - "out" - ] + else actx.call_loopy(_pt_t_unit_9, _in0=_pt_tmp_119, _in1=_pt_tmp_122)["out"] ) del _pt_tmp_340 _pt_tmp_338 = _pt_tmp_339 + _pt_tmp_341 del _pt_tmp_119, _pt_tmp_122 - _pt_tmp_337 = (0.5 + 0j) * _pt_tmp_338 + _pt_tmp_337 = 0.5 * _pt_tmp_338 del _pt_tmp_339, _pt_tmp_341 _pt_tmp_335 = _pt_tmp_336 + _pt_tmp_337 del _pt_tmp_338 _pt_tmp_334 = ( _pt_tmp_135 * _pt_tmp_335 if actx.supports_nonscalar_broadcasting - else actx.call_loopy(_pt_t_unit_12, _in0=_pt_tmp_135, _in1=_pt_tmp_335)[ - "out" - ] + else actx.call_loopy(_pt_t_unit_9, _in0=_pt_tmp_135, _in1=_pt_tmp_335)["out"] ) del _pt_tmp_336, _pt_tmp_337 _pt_tmp_331 = _pt_tmp_332 + _pt_tmp_334 del _pt_tmp_135 - _pt_tmp_330 = (0.5 + 0j) * _pt_tmp_331 + _pt_tmp_330 = 0.5 * _pt_tmp_331 del _pt_tmp_332, _pt_tmp_334 _pt_tmp_329 = ( _pt_tmp_330[_pt_tmp_136, _pt_tmp_82] if actx.permits_advanced_indexing else actx.call_loopy( - _pt_t_unit_13, in_0=_pt_tmp_330, in_1=_pt_tmp_136, in_2=_pt_tmp_82 + _pt_t_unit_10, in_0=_pt_tmp_330, in_1=_pt_tmp_136, in_2=_pt_tmp_82 )["out"] ) del _pt_tmp_331 _pt_tmp_328 = ( actx.np.where(_pt_tmp_89, _pt_tmp_329, 0) if actx.supports_nonscalar_broadcasting - else actx.call_loopy(_pt_t_unit_8, _in0=_pt_tmp_89, _in1=_pt_tmp_329)["out"] + else actx.call_loopy(_pt_t_unit_6, _in0=_pt_tmp_89, _in1=_pt_tmp_329)["out"] ) del _pt_tmp_330 _pt_tmp_328 = actx.tag_axis(0, (DiscretizationElementAxisTag(),), _pt_tmp_328) del _pt_tmp_329 _pt_tmp_328 = actx.tag_axis(1, (DiscretizationDOFAxisTag(),), _pt_tmp_328) - _pt_tmp_327 = 0j + _pt_tmp_328 - _pt_tmp_326 = 0j + _pt_tmp_327 + _pt_tmp_327 = 0.0 + _pt_tmp_328 + _pt_tmp_326 = 0.0 + _pt_tmp_327 del _pt_tmp_328 _pt_tmp_325 = _pt_tmp_326 + _pt_tmp_137 del _pt_tmp_327 @@ -1333,7 +1253,7 @@ def _rhs_inner( _pt_tmp_303 = actx.tag((FirstAxisIsElementsTag(),), _pt_tmp_303) del _pt_tmp_304 _pt_tmp_295 = _pt_tmp_296 - _pt_tmp_303 - _pt_tmp_294 = _pt_tmp_295 / np.complex128(1 + 0j) + _pt_tmp_294 = _pt_tmp_295 / np.float64(1.0) del _pt_tmp_296, _pt_tmp_303 _pt_tmp_348 = actx.einsum( "ij,ikl,jl->jk", _pt_tmp_146, _pt_data_0, _actx_in_1_1_0 @@ -1341,14 +1261,14 @@ def _rhs_inner( del _pt_tmp_295 _pt_tmp_348 = actx.tag((FirstAxisIsElementsTag(),), _pt_tmp_348) del _pt_tmp_146 - _pt_tmp_347 = (1 + 0j) * _pt_tmp_348 - _pt_tmp_346 = 0j + _pt_tmp_347 + _pt_tmp_347 = 1.0 * _pt_tmp_348 + _pt_tmp_346 = 0.0 + _pt_tmp_347 del _pt_tmp_348 _pt_tmp_350 = actx.einsum("ij,ikl,jl->jk", _pt_tmp_8, _pt_data_0, _actx_in_1_0_0) del _pt_tmp_347 _pt_tmp_350 = actx.tag((FirstAxisIsElementsTag(),), _pt_tmp_350) del _pt_tmp_8 - _pt_tmp_349 = (-1 + 0j) * _pt_tmp_350 + _pt_tmp_349 = -1.0 * _pt_tmp_350 _pt_tmp_345 = _pt_tmp_346 + _pt_tmp_349 del _pt_tmp_350 _pt_tmp_344 = -1 * _pt_tmp_345 @@ -1356,82 +1276,78 @@ def _rhs_inner( _pt_tmp_362 = ( _pt_tmp_47 * _pt_tmp_261 if actx.supports_nonscalar_broadcasting - else actx.call_loopy(_pt_t_unit_5, _in0=_pt_tmp_47, _in1=_pt_tmp_261)["out"] + else actx.call_loopy(_pt_t_unit_3, _in0=_pt_tmp_47, _in1=_pt_tmp_261)["out"] ) del _pt_tmp_345 - _pt_tmp_361 = 0j + _pt_tmp_362 + _pt_tmp_361 = 0.0 + _pt_tmp_362 del _pt_tmp_261, _pt_tmp_47 _pt_tmp_363 = ( _pt_tmp_54 * _pt_tmp_316 if actx.supports_nonscalar_broadcasting - else actx.call_loopy(_pt_t_unit_5, _in0=_pt_tmp_54, _in1=_pt_tmp_316)["out"] + else actx.call_loopy(_pt_t_unit_3, _in0=_pt_tmp_54, _in1=_pt_tmp_316)["out"] ) del _pt_tmp_362 _pt_tmp_360 = _pt_tmp_361 + _pt_tmp_363 del _pt_tmp_316, _pt_tmp_54 - _pt_tmp_359 = (0.5 + 0j) * _pt_tmp_360 + _pt_tmp_359 = 0.5 * _pt_tmp_360 del _pt_tmp_361, _pt_tmp_363 _pt_tmp_358 = ( _pt_tmp_359[_pt_tmp_81, _pt_tmp_82] if actx.permits_advanced_indexing else actx.call_loopy( - _pt_t_unit_7, in_0=_pt_tmp_359, in_1=_pt_tmp_81, in_2=_pt_tmp_82 + _pt_t_unit_5, in_0=_pt_tmp_359, in_1=_pt_tmp_81, in_2=_pt_tmp_82 )["out"] ) del _pt_tmp_360 _pt_tmp_357 = ( actx.np.where(_pt_tmp_25, _pt_tmp_358, 0) if actx.supports_nonscalar_broadcasting - else actx.call_loopy(_pt_t_unit_8, _in0=_pt_tmp_25, _in1=_pt_tmp_358)["out"] + else actx.call_loopy(_pt_t_unit_6, _in0=_pt_tmp_25, _in1=_pt_tmp_358)["out"] ) del _pt_tmp_359, _pt_tmp_81 _pt_tmp_357 = actx.tag_axis(0, (DiscretizationElementAxisTag(),), _pt_tmp_357) del _pt_tmp_25, _pt_tmp_358 _pt_tmp_357 = actx.tag_axis(1, (DiscretizationDOFAxisTag(),), _pt_tmp_357) - _pt_tmp_356 = 0j + _pt_tmp_357 - _pt_tmp_355 = 0j + _pt_tmp_356 + _pt_tmp_356 = 0.0 + _pt_tmp_357 + _pt_tmp_355 = 0.0 + _pt_tmp_356 del _pt_tmp_357 _pt_tmp_374 = ( _pt_tmp_107 * _pt_tmp_287 if actx.supports_nonscalar_broadcasting - else actx.call_loopy(_pt_t_unit_12, _in0=_pt_tmp_107, _in1=_pt_tmp_287)[ - "out" - ] + else actx.call_loopy(_pt_t_unit_9, _in0=_pt_tmp_107, _in1=_pt_tmp_287)["out"] ) del _pt_tmp_356 - _pt_tmp_373 = 0j + _pt_tmp_374 + _pt_tmp_373 = 0.0 + _pt_tmp_374 del _pt_tmp_107, _pt_tmp_287 _pt_tmp_375 = ( _pt_tmp_113 * _pt_tmp_335 if actx.supports_nonscalar_broadcasting - else actx.call_loopy(_pt_t_unit_12, _in0=_pt_tmp_113, _in1=_pt_tmp_335)[ - "out" - ] + else actx.call_loopy(_pt_t_unit_9, _in0=_pt_tmp_113, _in1=_pt_tmp_335)["out"] ) del _pt_tmp_374 _pt_tmp_372 = _pt_tmp_373 + _pt_tmp_375 del _pt_tmp_113, _pt_tmp_335 - _pt_tmp_371 = (0.5 + 0j) * _pt_tmp_372 + _pt_tmp_371 = 0.5 * _pt_tmp_372 del _pt_tmp_373, _pt_tmp_375 _pt_tmp_370 = ( _pt_tmp_371[_pt_tmp_136, _pt_tmp_82] if actx.permits_advanced_indexing else actx.call_loopy( - _pt_t_unit_13, in_0=_pt_tmp_371, in_1=_pt_tmp_136, in_2=_pt_tmp_82 + _pt_t_unit_10, in_0=_pt_tmp_371, in_1=_pt_tmp_136, in_2=_pt_tmp_82 )["out"] ) del _pt_tmp_372 _pt_tmp_369 = ( actx.np.where(_pt_tmp_89, _pt_tmp_370, 0) if actx.supports_nonscalar_broadcasting - else actx.call_loopy(_pt_t_unit_8, _in0=_pt_tmp_89, _in1=_pt_tmp_370)["out"] + else actx.call_loopy(_pt_t_unit_6, _in0=_pt_tmp_89, _in1=_pt_tmp_370)["out"] ) del _pt_tmp_136, _pt_tmp_371, _pt_tmp_82 _pt_tmp_369 = actx.tag_axis(0, (DiscretizationElementAxisTag(),), _pt_tmp_369) del _pt_tmp_370, _pt_tmp_89 _pt_tmp_369 = actx.tag_axis(1, (DiscretizationDOFAxisTag(),), _pt_tmp_369) - _pt_tmp_368 = 0j + _pt_tmp_369 - _pt_tmp_367 = 0j + _pt_tmp_368 + _pt_tmp_368 = 0.0 + _pt_tmp_369 + _pt_tmp_367 = 0.0 + _pt_tmp_368 del _pt_tmp_369 _pt_tmp_366 = _pt_tmp_367 + _pt_tmp_137 del _pt_tmp_368 @@ -1452,7 +1368,7 @@ def _rhs_inner( _pt_tmp_351 = actx.tag((FirstAxisIsElementsTag(),), _pt_tmp_351) del _pt_tmp_15, _pt_tmp_352 _pt_tmp_343 = _pt_tmp_344 - _pt_tmp_351 - _pt_tmp_342 = _pt_tmp_343 / np.complex128(1 + 0j) + _pt_tmp_342 = _pt_tmp_343 / np.float64(1.0) del _pt_tmp_344, _pt_tmp_351 _pt_tmp = make_obj_array( [_pt_tmp_0, _pt_tmp_138, _pt_tmp_197, _pt_tmp_232, _pt_tmp_294, _pt_tmp_342] diff --git a/actx_dgfem_suite/suite/tiny_maxwell_3D_P1/ref_input_args.pkl.dvc b/actx_dgfem_suite/suite/tiny_maxwell_3D_P1/ref_input_args.pkl.dvc index 9941519..3dd2ef2 100644 --- a/actx_dgfem_suite/suite/tiny_maxwell_3D_P1/ref_input_args.pkl.dvc +++ b/actx_dgfem_suite/suite/tiny_maxwell_3D_P1/ref_input_args.pkl.dvc @@ -1,5 +1,5 @@ outs: -- md5: d435376bb746886c0b6b47dea9646700 - size: 135895 +- md5: c5c4e77fd03ad88e47bf5900d6a7d778 + size: 74420 hash: md5 path: ref_input_args.pkl diff --git a/actx_dgfem_suite/suite/tiny_maxwell_3D_P1/ref_outputs.pkl.dvc b/actx_dgfem_suite/suite/tiny_maxwell_3D_P1/ref_outputs.pkl.dvc index 7905d57..ba3fd7f 100644 --- a/actx_dgfem_suite/suite/tiny_maxwell_3D_P1/ref_outputs.pkl.dvc +++ b/actx_dgfem_suite/suite/tiny_maxwell_3D_P1/ref_outputs.pkl.dvc @@ -1,5 +1,5 @@ outs: -- md5: ed09af9b81bbb692df3b0e6d8db23ed6 - size: 148122 +- md5: c4f0e67c84e903023db91c91505ec9eb + size: 74384 hash: md5 path: ref_outputs.pkl diff --git a/actx_dgfem_suite/suite/tiny_maxwell_3D_P2/literals.npz.dvc b/actx_dgfem_suite/suite/tiny_maxwell_3D_P2/literals.npz.dvc index 26b780a..139668e 100644 --- a/actx_dgfem_suite/suite/tiny_maxwell_3D_P2/literals.npz.dvc +++ b/actx_dgfem_suite/suite/tiny_maxwell_3D_P2/literals.npz.dvc @@ -1,5 +1,5 @@ outs: -- md5: 8edd2464e0719cfc94d5955fe6c45fea - size: 60788 +- md5: b3b56e32ce2a6105e793b870a7546fc5 + size: 58868 hash: md5 path: literals.npz diff --git a/actx_dgfem_suite/suite/tiny_maxwell_3D_P2/main.py b/actx_dgfem_suite/suite/tiny_maxwell_3D_P2/main.py index 3454478..b1616f3 100644 --- a/actx_dgfem_suite/suite/tiny_maxwell_3D_P2/main.py +++ b/actx_dgfem_suite/suite/tiny_maxwell_3D_P2/main.py @@ -58,62 +58,15 @@ def _rhs_inner( ], ) _pt_t_unit_10 = make_loopy_program( - "{ [_0, _1] : 0 <= _0 <= 107 and 0 <= _1 <= 5 }", - "out[_0, _1] = in_0[in_1[_0, 0] % 162, in_2[_0, _1] % 10]", - [ - lp.GlobalArg("out", dtype=np.float64, shape=(108, 6)), - lp.GlobalArg("in_0", dtype=np.float64, shape=(162, 10)), - lp.GlobalArg("in_1", dtype=np.int32, shape=(108, 1)), - lp.GlobalArg("in_2", dtype=np.int64, shape=(108, 6)), - ], - ) - _pt_t_unit_11 = make_loopy_program( - "{ [_0, _1] : 0 <= _0 <= 107 and 0 <= _1 <= 5 }", - "out[_0, _1] = in_0[in_1[_0, 0] % 162, in_2[_0, _1] % 10]", - [ - lp.GlobalArg("out", dtype=np.complex128, shape=(108, 6)), - lp.GlobalArg("in_0", dtype=np.complex128, shape=(162, 10)), - lp.GlobalArg("in_1", dtype=np.int32, shape=(108, 1)), - lp.GlobalArg("in_2", dtype=np.int64, shape=(108, 6)), - ], - ) - _pt_t_unit_12 = make_loopy_program( - "{ [_0, _1] : 0 <= _0 <= 107 and 0 <= _1 <= 5 }", - "out[_0, _1] = cast(np:dtype('complex128'), _in0[_0, 0])*_in1[_0, _1]", - [ - lp.GlobalArg("out", dtype=np.complex128, shape=(108, 6)), - lp.GlobalArg("_in0", dtype=np.float64, shape=(108, 1)), - lp.GlobalArg("_in1", dtype=np.complex128, shape=(108, 6)), - ], - ) - _pt_t_unit_13 = make_loopy_program( "{ [_0, _1] : 0 <= _0 <= 647 and 0 <= _1 <= 5 }", "out[_0, _1] = in_0[in_1[_0, 0] % 108, in_2[_0, _1] % 6]", [ - lp.GlobalArg("out", dtype=np.complex128, shape=(648, 6)), - lp.GlobalArg("in_0", dtype=np.complex128, shape=(108, 6)), + lp.GlobalArg("out", dtype=np.float64, shape=(648, 6)), + lp.GlobalArg("in_0", dtype=np.float64, shape=(108, 6)), lp.GlobalArg("in_1", dtype=np.int32, shape=(648, 1)), lp.GlobalArg("in_2", dtype=np.int64, shape=(648, 6)), ], ) - _pt_t_unit_14 = make_loopy_program( - "{ [_0, _1] : 0 <= _0 <= 539 and 0 <= _1 <= 5 }", - "out[_0, _1] = _in0[_0, 0]*_in1[_0, _1]", - [ - lp.GlobalArg("out", dtype=np.float64, shape=(540, 6)), - lp.GlobalArg("_in0", dtype=np.float64, shape=(540, 1)), - lp.GlobalArg("_in1", dtype=np.float64, shape=(540, 6)), - ], - ) - _pt_t_unit_15 = make_loopy_program( - "{ [_0, _1] : 0 <= _0 <= 107 and 0 <= _1 <= 5 }", - "out[_0, _1] = _in0[_0, 0]*_in1[_0, _1]", - [ - lp.GlobalArg("out", dtype=np.float64, shape=(108, 6)), - lp.GlobalArg("_in0", dtype=np.float64, shape=(108, 1)), - lp.GlobalArg("_in1", dtype=np.float64, shape=(108, 6)), - ], - ) _pt_t_unit_2 = make_loopy_program( "{ [_0, _1] : 0 <= _0 <= 539 and 0 <= _1 <= 5 }", "out[_0, _1] = in_0[in_1[_0, 0] % 540, in_2[_0, _1] % 6]", @@ -126,34 +79,14 @@ def _rhs_inner( ) _pt_t_unit_3 = make_loopy_program( "{ [_0, _1] : 0 <= _0 <= 539 and 0 <= _1 <= 5 }", - "out[_0, _1] = in_0[in_1[_0, 0] % 162, in_2[_0, _1] % 10]", - [ - lp.GlobalArg("out", dtype=np.complex128, shape=(540, 6)), - lp.GlobalArg("in_0", dtype=np.complex128, shape=(162, 10)), - lp.GlobalArg("in_1", dtype=np.int32, shape=(540, 1)), - lp.GlobalArg("in_2", dtype=np.int64, shape=(540, 6)), - ], - ) - _pt_t_unit_4 = make_loopy_program( - "{ [_0, _1] : 0 <= _0 <= 539 and 0 <= _1 <= 5 }", - "out[_0, _1] = in_0[in_1[_0, 0] % 540, in_2[_0, _1] % 6]", - [ - lp.GlobalArg("out", dtype=np.complex128, shape=(540, 6)), - lp.GlobalArg("in_0", dtype=np.complex128, shape=(540, 6)), - lp.GlobalArg("in_1", dtype=np.int32, shape=(540, 1)), - lp.GlobalArg("in_2", dtype=np.int64, shape=(540, 6)), - ], - ) - _pt_t_unit_5 = make_loopy_program( - "{ [_0, _1] : 0 <= _0 <= 539 and 0 <= _1 <= 5 }", - "out[_0, _1] = cast(np:dtype('complex128'), _in0[_0, 0])*_in1[_0, _1]", + "out[_0, _1] = _in0[_0, 0]*_in1[_0, _1]", [ - lp.GlobalArg("out", dtype=np.complex128, shape=(540, 6)), + lp.GlobalArg("out", dtype=np.float64, shape=(540, 6)), lp.GlobalArg("_in0", dtype=np.float64, shape=(540, 1)), - lp.GlobalArg("_in1", dtype=np.complex128, shape=(540, 6)), + lp.GlobalArg("_in1", dtype=np.float64, shape=(540, 6)), ], ) - _pt_t_unit_6 = make_loopy_program( + _pt_t_unit_4 = make_loopy_program( "{ [_0, _1] : 0 <= _0 <= 647 and 0 <= _1 <= 5 }", "out[_0, _1] = in_0[in_1[_0] % 1, _1]", [ @@ -162,26 +95,26 @@ def _rhs_inner( lp.GlobalArg("in_1", dtype=np.int8, shape=(648,)), ], ) - _pt_t_unit_7 = make_loopy_program( + _pt_t_unit_5 = make_loopy_program( "{ [_0, _1] : 0 <= _0 <= 647 and 0 <= _1 <= 5 }", "out[_0, _1] = in_0[in_1[_0, 0] % 540, in_2[_0, _1] % 6]", [ - lp.GlobalArg("out", dtype=np.complex128, shape=(648, 6)), - lp.GlobalArg("in_0", dtype=np.complex128, shape=(540, 6)), + lp.GlobalArg("out", dtype=np.float64, shape=(648, 6)), + lp.GlobalArg("in_0", dtype=np.float64, shape=(540, 6)), lp.GlobalArg("in_1", dtype=np.int32, shape=(648, 1)), lp.GlobalArg("in_2", dtype=np.int64, shape=(648, 6)), ], ) - _pt_t_unit_8 = make_loopy_program( + _pt_t_unit_6 = make_loopy_program( "{ [_0, _1] : 0 <= _0 <= 647 and 0 <= _1 <= 5 }", "out[_0, _1] = _in1[_0, _1] if _in0[_0, 0] else 0", [ - lp.GlobalArg("out", dtype=np.complex128, shape=(648, 6)), + lp.GlobalArg("out", dtype=np.float64, shape=(648, 6)), lp.GlobalArg("_in0", dtype=np.int8, shape=(648, 1)), - lp.GlobalArg("_in1", dtype=np.complex128, shape=(648, 6)), + lp.GlobalArg("_in1", dtype=np.float64, shape=(648, 6)), ], ) - _pt_t_unit_9 = make_loopy_program( + _pt_t_unit_7 = make_loopy_program( "{ [_0, _1] : 0 <= _0 <= 107 and 0 <= _1 <= 5 }", "out[_0, _1] = in_0[in_1[_0] % 4, _1]", [ @@ -190,6 +123,25 @@ def _rhs_inner( lp.GlobalArg("in_1", dtype=np.int8, shape=(108,)), ], ) + _pt_t_unit_8 = make_loopy_program( + "{ [_0, _1] : 0 <= _0 <= 107 and 0 <= _1 <= 5 }", + "out[_0, _1] = in_0[in_1[_0, 0] % 162, in_2[_0, _1] % 10]", + [ + lp.GlobalArg("out", dtype=np.float64, shape=(108, 6)), + lp.GlobalArg("in_0", dtype=np.float64, shape=(162, 10)), + lp.GlobalArg("in_1", dtype=np.int32, shape=(108, 1)), + lp.GlobalArg("in_2", dtype=np.int64, shape=(108, 6)), + ], + ) + _pt_t_unit_9 = make_loopy_program( + "{ [_0, _1] : 0 <= _0 <= 107 and 0 <= _1 <= 5 }", + "out[_0, _1] = _in0[_0, 0]*_in1[_0, _1]", + [ + lp.GlobalArg("out", dtype=np.float64, shape=(108, 6)), + lp.GlobalArg("_in0", dtype=np.float64, shape=(108, 1)), + lp.GlobalArg("_in1", dtype=np.float64, shape=(108, 6)), + ], + ) _pt_data = actx.thaw(npzfile["_pt_data"]) _pt_tmp_9 = _pt_data[1] _pt_tmp_8 = _pt_tmp_9[:, :, 0] @@ -206,10 +158,10 @@ def _rhs_inner( _pt_tmp_11 = actx.einsum("ij,ikl,jl->jk", _pt_tmp_12, _pt_data_0, _actx_in_1_4_0) del _pt_tmp_13 _pt_tmp_11 = actx.tag((FirstAxisIsElementsTag(),), _pt_tmp_11) - _pt_tmp_10 = (-1 + 0j) * _pt_tmp_11 + _pt_tmp_10 = -1.0 * _pt_tmp_11 _pt_tmp_4 = _pt_tmp_5 + _pt_tmp_10 del _pt_tmp_11 - _pt_tmp_3 = 0j - _pt_tmp_4 + _pt_tmp_3 = 0.0 - _pt_tmp_4 del _pt_tmp_10, _pt_tmp_5 _pt_tmp_2 = -1 * _pt_tmp_3 del _pt_tmp_4 @@ -278,75 +230,75 @@ def _rhs_inner( _actx_in_1_1_0[_pt_tmp_39, _pt_tmp_40] if actx.permits_advanced_indexing else actx.call_loopy( - _pt_t_unit_3, in_0=_actx_in_1_1_0, in_1=_pt_tmp_39, in_2=_pt_tmp_40 + _pt_t_unit_0, in_0=_actx_in_1_1_0, in_1=_pt_tmp_39, in_2=_pt_tmp_40 )["out"] ) _pt_tmp_52 = actx.tag_axis(0, (DiscretizationElementAxisTag(),), _pt_tmp_52) _pt_tmp_52 = actx.tag_axis(1, (DiscretizationDOFAxisTag(),), _pt_tmp_52) - _pt_tmp_51 = 0j + _pt_tmp_52 + _pt_tmp_51 = 0.0 + _pt_tmp_52 _pt_tmp_50 = ( _pt_tmp_51[_pt_tmp_41, _pt_tmp_42] if actx.permits_advanced_indexing else actx.call_loopy( - _pt_t_unit_4, in_0=_pt_tmp_51, in_1=_pt_tmp_41, in_2=_pt_tmp_42 + _pt_t_unit_2, in_0=_pt_tmp_51, in_1=_pt_tmp_41, in_2=_pt_tmp_42 )["out"] ) del _pt_tmp_52 _pt_tmp_50 = actx.tag_axis(0, (DiscretizationElementAxisTag(),), _pt_tmp_50) _pt_tmp_50 = actx.tag_axis(1, (DiscretizationDOFAxisTag(),), _pt_tmp_50) - _pt_tmp_49 = 0j + _pt_tmp_50 + _pt_tmp_49 = 0.0 + _pt_tmp_50 _pt_tmp_48 = _pt_tmp_49 - _pt_tmp_51 del _pt_tmp_50 _pt_tmp_46 = ( _pt_tmp_47 * _pt_tmp_48 if actx.supports_nonscalar_broadcasting - else actx.call_loopy(_pt_t_unit_5, _in0=_pt_tmp_47, _in1=_pt_tmp_48)["out"] + else actx.call_loopy(_pt_t_unit_3, _in0=_pt_tmp_47, _in1=_pt_tmp_48)["out"] ) del _pt_tmp_49, _pt_tmp_51 - _pt_tmp_45 = 0j + _pt_tmp_46 + _pt_tmp_45 = 0.0 + _pt_tmp_46 _pt_tmp_54 = -1.0 * _pt_data_6 del _pt_tmp_46 _pt_tmp_59 = ( _actx_in_1_0_0[_pt_tmp_39, _pt_tmp_40] if actx.permits_advanced_indexing else actx.call_loopy( - _pt_t_unit_3, in_0=_actx_in_1_0_0, in_1=_pt_tmp_39, in_2=_pt_tmp_40 + _pt_t_unit_0, in_0=_actx_in_1_0_0, in_1=_pt_tmp_39, in_2=_pt_tmp_40 )["out"] ) _pt_tmp_59 = actx.tag_axis(0, (DiscretizationElementAxisTag(),), _pt_tmp_59) _pt_tmp_59 = actx.tag_axis(1, (DiscretizationDOFAxisTag(),), _pt_tmp_59) - _pt_tmp_58 = 0j + _pt_tmp_59 + _pt_tmp_58 = 0.0 + _pt_tmp_59 _pt_tmp_57 = ( _pt_tmp_58[_pt_tmp_41, _pt_tmp_42] if actx.permits_advanced_indexing else actx.call_loopy( - _pt_t_unit_4, in_0=_pt_tmp_58, in_1=_pt_tmp_41, in_2=_pt_tmp_42 + _pt_t_unit_2, in_0=_pt_tmp_58, in_1=_pt_tmp_41, in_2=_pt_tmp_42 )["out"] ) del _pt_tmp_59 _pt_tmp_57 = actx.tag_axis(0, (DiscretizationElementAxisTag(),), _pt_tmp_57) _pt_tmp_57 = actx.tag_axis(1, (DiscretizationDOFAxisTag(),), _pt_tmp_57) - _pt_tmp_56 = 0j + _pt_tmp_57 + _pt_tmp_56 = 0.0 + _pt_tmp_57 _pt_tmp_55 = _pt_tmp_56 - _pt_tmp_58 del _pt_tmp_57 _pt_tmp_53 = ( _pt_tmp_54 * _pt_tmp_55 if actx.supports_nonscalar_broadcasting - else actx.call_loopy(_pt_t_unit_5, _in0=_pt_tmp_54, _in1=_pt_tmp_55)["out"] + else actx.call_loopy(_pt_t_unit_3, _in0=_pt_tmp_54, _in1=_pt_tmp_55)["out"] ) del _pt_tmp_56, _pt_tmp_58 _pt_tmp_44 = _pt_tmp_45 + _pt_tmp_53 - _pt_tmp_43 = (0.5 + 0j) * _pt_tmp_44 + _pt_tmp_43 = 0.5 * _pt_tmp_44 del _pt_tmp_45, _pt_tmp_53 _pt_tmp_32 = _pt_tmp_33 - _pt_tmp_43 del _pt_tmp_44 _pt_tmp_30 = ( _pt_tmp_31 * _pt_tmp_32 if actx.supports_nonscalar_broadcasting - else actx.call_loopy(_pt_t_unit_5, _in0=_pt_tmp_31, _in1=_pt_tmp_32)["out"] + else actx.call_loopy(_pt_t_unit_3, _in0=_pt_tmp_31, _in1=_pt_tmp_32)["out"] ) del _pt_tmp_33, _pt_tmp_43 - _pt_tmp_29 = 0j + _pt_tmp_30 + _pt_tmp_29 = 0.0 + _pt_tmp_30 _pt_data_14 = actx.thaw(npzfile["_pt_data_14"]) del _pt_tmp_30 _pt_tmp_61 = -1.0 * _pt_data_14 @@ -354,78 +306,78 @@ def _rhs_inner( _actx_in_1_4_0[_pt_tmp_39, _pt_tmp_40] if actx.permits_advanced_indexing else actx.call_loopy( - _pt_t_unit_3, in_0=_actx_in_1_4_0, in_1=_pt_tmp_39, in_2=_pt_tmp_40 + _pt_t_unit_0, in_0=_actx_in_1_4_0, in_1=_pt_tmp_39, in_2=_pt_tmp_40 )["out"] ) _pt_tmp_68 = actx.tag_axis(0, (DiscretizationElementAxisTag(),), _pt_tmp_68) _pt_tmp_68 = actx.tag_axis(1, (DiscretizationDOFAxisTag(),), _pt_tmp_68) - _pt_tmp_67 = 0j + _pt_tmp_68 + _pt_tmp_67 = 0.0 + _pt_tmp_68 _pt_tmp_66 = ( _pt_tmp_67[_pt_tmp_41, _pt_tmp_42] if actx.permits_advanced_indexing else actx.call_loopy( - _pt_t_unit_4, in_0=_pt_tmp_67, in_1=_pt_tmp_41, in_2=_pt_tmp_42 + _pt_t_unit_2, in_0=_pt_tmp_67, in_1=_pt_tmp_41, in_2=_pt_tmp_42 )["out"] ) del _pt_tmp_68 _pt_tmp_66 = actx.tag_axis(0, (DiscretizationElementAxisTag(),), _pt_tmp_66) _pt_tmp_66 = actx.tag_axis(1, (DiscretizationDOFAxisTag(),), _pt_tmp_66) - _pt_tmp_65 = 0j + _pt_tmp_66 + _pt_tmp_65 = 0.0 + _pt_tmp_66 _pt_tmp_64 = _pt_tmp_65 - _pt_tmp_67 del _pt_tmp_66 - _pt_tmp_63 = (1 + 0j) * _pt_tmp_64 + _pt_tmp_63 = 1.0 * _pt_tmp_64 del _pt_tmp_65, _pt_tmp_67 _pt_tmp_73 = -1.0 * _pt_data_13 _pt_tmp_78 = ( _actx_in_1_2_0[_pt_tmp_39, _pt_tmp_40] if actx.permits_advanced_indexing else actx.call_loopy( - _pt_t_unit_3, in_0=_actx_in_1_2_0, in_1=_pt_tmp_39, in_2=_pt_tmp_40 + _pt_t_unit_0, in_0=_actx_in_1_2_0, in_1=_pt_tmp_39, in_2=_pt_tmp_40 )["out"] ) _pt_tmp_78 = actx.tag_axis(0, (DiscretizationElementAxisTag(),), _pt_tmp_78) _pt_tmp_78 = actx.tag_axis(1, (DiscretizationDOFAxisTag(),), _pt_tmp_78) - _pt_tmp_77 = 0j + _pt_tmp_78 + _pt_tmp_77 = 0.0 + _pt_tmp_78 _pt_tmp_76 = ( _pt_tmp_77[_pt_tmp_41, _pt_tmp_42] if actx.permits_advanced_indexing else actx.call_loopy( - _pt_t_unit_4, in_0=_pt_tmp_77, in_1=_pt_tmp_41, in_2=_pt_tmp_42 + _pt_t_unit_2, in_0=_pt_tmp_77, in_1=_pt_tmp_41, in_2=_pt_tmp_42 )["out"] ) del _pt_tmp_78 _pt_tmp_76 = actx.tag_axis(0, (DiscretizationElementAxisTag(),), _pt_tmp_76) _pt_tmp_76 = actx.tag_axis(1, (DiscretizationDOFAxisTag(),), _pt_tmp_76) - _pt_tmp_75 = 0j + _pt_tmp_76 + _pt_tmp_75 = 0.0 + _pt_tmp_76 _pt_tmp_74 = _pt_tmp_75 - _pt_tmp_77 del _pt_tmp_76 _pt_tmp_72 = ( _pt_tmp_73 * _pt_tmp_74 if actx.supports_nonscalar_broadcasting - else actx.call_loopy(_pt_t_unit_5, _in0=_pt_tmp_73, _in1=_pt_tmp_74)["out"] + else actx.call_loopy(_pt_t_unit_3, _in0=_pt_tmp_73, _in1=_pt_tmp_74)["out"] ) del _pt_tmp_75, _pt_tmp_77 - _pt_tmp_71 = 0j + _pt_tmp_72 + _pt_tmp_71 = 0.0 + _pt_tmp_72 _pt_tmp_80 = 1.0 * _pt_data_14 del _pt_tmp_72 _pt_tmp_79 = ( _pt_tmp_80 * _pt_tmp_55 if actx.supports_nonscalar_broadcasting - else actx.call_loopy(_pt_t_unit_5, _in0=_pt_tmp_80, _in1=_pt_tmp_55)["out"] + else actx.call_loopy(_pt_t_unit_3, _in0=_pt_tmp_80, _in1=_pt_tmp_55)["out"] ) _pt_tmp_70 = _pt_tmp_71 + _pt_tmp_79 - _pt_tmp_69 = (0.5 + 0j) * _pt_tmp_70 + _pt_tmp_69 = 0.5 * _pt_tmp_70 del _pt_tmp_71, _pt_tmp_79 _pt_tmp_62 = _pt_tmp_63 - _pt_tmp_69 del _pt_tmp_70 _pt_tmp_60 = ( _pt_tmp_61 * _pt_tmp_62 if actx.supports_nonscalar_broadcasting - else actx.call_loopy(_pt_t_unit_5, _in0=_pt_tmp_61, _in1=_pt_tmp_62)["out"] + else actx.call_loopy(_pt_t_unit_3, _in0=_pt_tmp_61, _in1=_pt_tmp_62)["out"] ) del _pt_tmp_63, _pt_tmp_69 _pt_tmp_28 = _pt_tmp_29 + _pt_tmp_60 - _pt_tmp_27 = (-0.5 + 0j) * _pt_tmp_28 + _pt_tmp_27 = -0.5 * _pt_tmp_28 del _pt_tmp_29, _pt_tmp_60 _pt_data_15 = actx.thaw(npzfile["_pt_data_15"]) del _pt_tmp_28 @@ -435,26 +387,26 @@ def _rhs_inner( _pt_tmp_82 = ( _pt_data_16[_pt_data_17] if actx.permits_advanced_indexing - else actx.call_loopy(_pt_t_unit_6, in_0=_pt_data_16, in_1=_pt_data_17)["out"] + else actx.call_loopy(_pt_t_unit_4, in_0=_pt_data_16, in_1=_pt_data_17)["out"] ) _pt_tmp_26 = ( _pt_tmp_27[_pt_tmp_81, _pt_tmp_82] if actx.permits_advanced_indexing else actx.call_loopy( - _pt_t_unit_7, in_0=_pt_tmp_27, in_1=_pt_tmp_81, in_2=_pt_tmp_82 + _pt_t_unit_5, in_0=_pt_tmp_27, in_1=_pt_tmp_81, in_2=_pt_tmp_82 )["out"] ) _pt_tmp_24 = ( actx.np.where(_pt_tmp_25, _pt_tmp_26, 0) if actx.supports_nonscalar_broadcasting - else actx.call_loopy(_pt_t_unit_8, _in0=_pt_tmp_25, _in1=_pt_tmp_26)["out"] + else actx.call_loopy(_pt_t_unit_6, _in0=_pt_tmp_25, _in1=_pt_tmp_26)["out"] ) del _pt_tmp_27 _pt_tmp_24 = actx.tag_axis(0, (DiscretizationElementAxisTag(),), _pt_tmp_24) del _pt_tmp_26 _pt_tmp_24 = actx.tag_axis(1, (DiscretizationDOFAxisTag(),), _pt_tmp_24) - _pt_tmp_23 = 0j + _pt_tmp_24 - _pt_tmp_22 = 0j + _pt_tmp_23 + _pt_tmp_23 = 0.0 + _pt_tmp_24 + _pt_tmp_22 = 0.0 + _pt_tmp_23 del _pt_tmp_24 _pt_data_18 = actx.thaw(npzfile["_pt_data_18"]) del _pt_tmp_23 @@ -467,13 +419,13 @@ def _rhs_inner( _pt_tmp_102 = ( _pt_data_8[_pt_data_21] if actx.permits_advanced_indexing - else actx.call_loopy(_pt_t_unit_9, in_0=_pt_data_8, in_1=_pt_data_21)["out"] + else actx.call_loopy(_pt_t_unit_7, in_0=_pt_data_8, in_1=_pt_data_21)["out"] ) _pt_tmp_100 = ( _actx_in_1_5_0[_pt_tmp_101, _pt_tmp_102] if actx.permits_advanced_indexing else actx.call_loopy( - _pt_t_unit_10, in_0=_actx_in_1_5_0, in_1=_pt_tmp_101, in_2=_pt_tmp_102 + _pt_t_unit_8, in_0=_actx_in_1_5_0, in_1=_pt_tmp_101, in_2=_pt_tmp_102 )["out"] ) _pt_tmp_100 = actx.tag_axis(0, (DiscretizationElementAxisTag(),), _pt_tmp_100) @@ -489,59 +441,55 @@ def _rhs_inner( _actx_in_1_1_0[_pt_tmp_101, _pt_tmp_102] if actx.permits_advanced_indexing else actx.call_loopy( - _pt_t_unit_11, in_0=_actx_in_1_1_0, in_1=_pt_tmp_101, in_2=_pt_tmp_102 + _pt_t_unit_8, in_0=_actx_in_1_1_0, in_1=_pt_tmp_101, in_2=_pt_tmp_102 )["out"] ) _pt_tmp_111 = actx.tag_axis(0, (DiscretizationElementAxisTag(),), _pt_tmp_111) _pt_tmp_111 = actx.tag_axis(1, (DiscretizationDOFAxisTag(),), _pt_tmp_111) - _pt_tmp_110 = 0j + _pt_tmp_111 + _pt_tmp_110 = 0.0 + _pt_tmp_111 _pt_tmp_109 = -1 * _pt_tmp_110 del _pt_tmp_111 _pt_tmp_108 = _pt_tmp_109 - _pt_tmp_110 _pt_tmp_106 = ( _pt_tmp_107 * _pt_tmp_108 if actx.supports_nonscalar_broadcasting - else actx.call_loopy(_pt_t_unit_12, _in0=_pt_tmp_107, _in1=_pt_tmp_108)[ - "out" - ] + else actx.call_loopy(_pt_t_unit_9, _in0=_pt_tmp_107, _in1=_pt_tmp_108)["out"] ) del _pt_tmp_109, _pt_tmp_110 - _pt_tmp_105 = 0j + _pt_tmp_106 + _pt_tmp_105 = 0.0 + _pt_tmp_106 _pt_tmp_113 = -1.0 * _pt_data_19 del _pt_tmp_106 _pt_tmp_117 = ( _actx_in_1_0_0[_pt_tmp_101, _pt_tmp_102] if actx.permits_advanced_indexing else actx.call_loopy( - _pt_t_unit_11, in_0=_actx_in_1_0_0, in_1=_pt_tmp_101, in_2=_pt_tmp_102 + _pt_t_unit_8, in_0=_actx_in_1_0_0, in_1=_pt_tmp_101, in_2=_pt_tmp_102 )["out"] ) _pt_tmp_117 = actx.tag_axis(0, (DiscretizationElementAxisTag(),), _pt_tmp_117) _pt_tmp_117 = actx.tag_axis(1, (DiscretizationDOFAxisTag(),), _pt_tmp_117) - _pt_tmp_116 = 0j + _pt_tmp_117 + _pt_tmp_116 = 0.0 + _pt_tmp_117 _pt_tmp_115 = -1 * _pt_tmp_116 del _pt_tmp_117 _pt_tmp_114 = _pt_tmp_115 - _pt_tmp_116 _pt_tmp_112 = ( _pt_tmp_113 * _pt_tmp_114 if actx.supports_nonscalar_broadcasting - else actx.call_loopy(_pt_t_unit_12, _in0=_pt_tmp_113, _in1=_pt_tmp_114)[ - "out" - ] + else actx.call_loopy(_pt_t_unit_9, _in0=_pt_tmp_113, _in1=_pt_tmp_114)["out"] ) del _pt_tmp_115, _pt_tmp_116 _pt_tmp_104 = _pt_tmp_105 + _pt_tmp_112 - _pt_tmp_103 = (0.5 + 0j) * _pt_tmp_104 + _pt_tmp_103 = 0.5 * _pt_tmp_104 del _pt_tmp_105, _pt_tmp_112 _pt_tmp_96 = _pt_tmp_97 - _pt_tmp_103 del _pt_tmp_104 _pt_tmp_94 = ( _pt_tmp_95 * _pt_tmp_96 if actx.supports_nonscalar_broadcasting - else actx.call_loopy(_pt_t_unit_12, _in0=_pt_tmp_95, _in1=_pt_tmp_96)["out"] + else actx.call_loopy(_pt_t_unit_9, _in0=_pt_tmp_95, _in1=_pt_tmp_96)["out"] ) del _pt_tmp_103, _pt_tmp_97 - _pt_tmp_93 = 0j + _pt_tmp_94 + _pt_tmp_93 = 0.0 + _pt_tmp_94 _pt_data_23 = actx.thaw(npzfile["_pt_data_23"]) del _pt_tmp_94 _pt_tmp_119 = -1.0 * _pt_data_23 @@ -549,63 +497,57 @@ def _rhs_inner( _actx_in_1_4_0[_pt_tmp_101, _pt_tmp_102] if actx.permits_advanced_indexing else actx.call_loopy( - _pt_t_unit_11, in_0=_actx_in_1_4_0, in_1=_pt_tmp_101, in_2=_pt_tmp_102 + _pt_t_unit_8, in_0=_actx_in_1_4_0, in_1=_pt_tmp_101, in_2=_pt_tmp_102 )["out"] ) _pt_tmp_124 = actx.tag_axis(0, (DiscretizationElementAxisTag(),), _pt_tmp_124) _pt_tmp_124 = actx.tag_axis(1, (DiscretizationDOFAxisTag(),), _pt_tmp_124) - _pt_tmp_123 = 0j + _pt_tmp_124 + _pt_tmp_123 = 0.0 + _pt_tmp_124 _pt_tmp_122 = _pt_tmp_123 - _pt_tmp_123 del _pt_tmp_124 - _pt_tmp_121 = (1 + 0j) * _pt_tmp_122 + _pt_tmp_121 = 1.0 * _pt_tmp_122 del _pt_tmp_123 _pt_tmp_129 = -1.0 * _pt_data_22 _pt_tmp_133 = ( _actx_in_1_2_0[_pt_tmp_101, _pt_tmp_102] if actx.permits_advanced_indexing else actx.call_loopy( - _pt_t_unit_11, in_0=_actx_in_1_2_0, in_1=_pt_tmp_101, in_2=_pt_tmp_102 + _pt_t_unit_8, in_0=_actx_in_1_2_0, in_1=_pt_tmp_101, in_2=_pt_tmp_102 )["out"] ) _pt_tmp_133 = actx.tag_axis(0, (DiscretizationElementAxisTag(),), _pt_tmp_133) _pt_tmp_133 = actx.tag_axis(1, (DiscretizationDOFAxisTag(),), _pt_tmp_133) - _pt_tmp_132 = 0j + _pt_tmp_133 + _pt_tmp_132 = 0.0 + _pt_tmp_133 _pt_tmp_131 = -1 * _pt_tmp_132 del _pt_tmp_133 _pt_tmp_130 = _pt_tmp_131 - _pt_tmp_132 _pt_tmp_128 = ( _pt_tmp_129 * _pt_tmp_130 if actx.supports_nonscalar_broadcasting - else actx.call_loopy(_pt_t_unit_12, _in0=_pt_tmp_129, _in1=_pt_tmp_130)[ - "out" - ] + else actx.call_loopy(_pt_t_unit_9, _in0=_pt_tmp_129, _in1=_pt_tmp_130)["out"] ) del _pt_tmp_131, _pt_tmp_132 - _pt_tmp_127 = 0j + _pt_tmp_128 + _pt_tmp_127 = 0.0 + _pt_tmp_128 _pt_tmp_135 = 1.0 * _pt_data_23 del _pt_tmp_128 _pt_tmp_134 = ( _pt_tmp_135 * _pt_tmp_114 if actx.supports_nonscalar_broadcasting - else actx.call_loopy(_pt_t_unit_12, _in0=_pt_tmp_135, _in1=_pt_tmp_114)[ - "out" - ] + else actx.call_loopy(_pt_t_unit_9, _in0=_pt_tmp_135, _in1=_pt_tmp_114)["out"] ) _pt_tmp_126 = _pt_tmp_127 + _pt_tmp_134 - _pt_tmp_125 = (0.5 + 0j) * _pt_tmp_126 + _pt_tmp_125 = 0.5 * _pt_tmp_126 del _pt_tmp_127, _pt_tmp_134 _pt_tmp_120 = _pt_tmp_121 - _pt_tmp_125 del _pt_tmp_126 _pt_tmp_118 = ( _pt_tmp_119 * _pt_tmp_120 if actx.supports_nonscalar_broadcasting - else actx.call_loopy(_pt_t_unit_12, _in0=_pt_tmp_119, _in1=_pt_tmp_120)[ - "out" - ] + else actx.call_loopy(_pt_t_unit_9, _in0=_pt_tmp_119, _in1=_pt_tmp_120)["out"] ) del _pt_tmp_121, _pt_tmp_125 _pt_tmp_92 = _pt_tmp_93 + _pt_tmp_118 - _pt_tmp_91 = (-0.5 + 0j) * _pt_tmp_92 + _pt_tmp_91 = -0.5 * _pt_tmp_92 del _pt_tmp_118, _pt_tmp_93 _pt_data_24 = actx.thaw(npzfile["_pt_data_24"]) del _pt_tmp_92 @@ -614,22 +556,22 @@ def _rhs_inner( _pt_tmp_91[_pt_tmp_136, _pt_tmp_82] if actx.permits_advanced_indexing else actx.call_loopy( - _pt_t_unit_13, in_0=_pt_tmp_91, in_1=_pt_tmp_136, in_2=_pt_tmp_82 + _pt_t_unit_10, in_0=_pt_tmp_91, in_1=_pt_tmp_136, in_2=_pt_tmp_82 )["out"] ) _pt_tmp_88 = ( actx.np.where(_pt_tmp_89, _pt_tmp_90, 0) if actx.supports_nonscalar_broadcasting - else actx.call_loopy(_pt_t_unit_8, _in0=_pt_tmp_89, _in1=_pt_tmp_90)["out"] + else actx.call_loopy(_pt_t_unit_6, _in0=_pt_tmp_89, _in1=_pt_tmp_90)["out"] ) del _pt_tmp_91 _pt_tmp_88 = actx.tag_axis(0, (DiscretizationElementAxisTag(),), _pt_tmp_88) del _pt_tmp_90 _pt_tmp_88 = actx.tag_axis(1, (DiscretizationDOFAxisTag(),), _pt_tmp_88) - _pt_tmp_87 = 0j + _pt_tmp_88 - _pt_tmp_86 = 0j + _pt_tmp_87 + _pt_tmp_87 = 0.0 + _pt_tmp_88 + _pt_tmp_86 = 0.0 + _pt_tmp_87 del _pt_tmp_88 - _pt_tmp_137 = actx.np.zeros((648, 6), dtype=np.complex128) + _pt_tmp_137 = actx.np.zeros((648, 6), dtype=np.float64) del _pt_tmp_87 _pt_tmp_85 = _pt_tmp_86 + _pt_tmp_137 _pt_tmp_84 = _pt_tmp_85 + _pt_tmp_137 @@ -649,7 +591,7 @@ def _rhs_inner( _pt_tmp_14 = actx.tag((FirstAxisIsElementsTag(),), _pt_tmp_14) del _pt_tmp_17 _pt_tmp_1 = _pt_tmp_2 - _pt_tmp_14 - _pt_tmp_0 = _pt_tmp_1 / np.complex128(1 + 0j) + _pt_tmp_0 = _pt_tmp_1 / np.float64(1.0) del _pt_tmp_14, _pt_tmp_2 _pt_tmp_147 = _pt_data[0] del _pt_tmp_1 @@ -667,168 +609,164 @@ def _rhs_inner( ) del _pt_tmp_144 _pt_tmp_149 = actx.tag((FirstAxisIsElementsTag(),), _pt_tmp_149) - _pt_tmp_148 = (1 + 0j) * _pt_tmp_149 + _pt_tmp_148 = 1.0 * _pt_tmp_149 _pt_tmp_142 = _pt_tmp_143 + _pt_tmp_148 del _pt_tmp_149 - _pt_tmp_141 = 0j - _pt_tmp_142 + _pt_tmp_141 = 0.0 - _pt_tmp_142 del _pt_tmp_143, _pt_tmp_148 _pt_tmp_140 = -1 * _pt_tmp_141 del _pt_tmp_142 _pt_tmp_161 = ( _pt_tmp_73 * _pt_tmp_32 if actx.supports_nonscalar_broadcasting - else actx.call_loopy(_pt_t_unit_5, _in0=_pt_tmp_73, _in1=_pt_tmp_32)["out"] + else actx.call_loopy(_pt_t_unit_3, _in0=_pt_tmp_73, _in1=_pt_tmp_32)["out"] ) del _pt_tmp_141 - _pt_tmp_160 = 0j + _pt_tmp_161 + _pt_tmp_160 = 0.0 + _pt_tmp_161 del _pt_tmp_32 _pt_tmp_169 = ( _actx_in_1_3_0[_pt_tmp_39, _pt_tmp_40] if actx.permits_advanced_indexing else actx.call_loopy( - _pt_t_unit_3, in_0=_actx_in_1_3_0, in_1=_pt_tmp_39, in_2=_pt_tmp_40 + _pt_t_unit_0, in_0=_actx_in_1_3_0, in_1=_pt_tmp_39, in_2=_pt_tmp_40 )["out"] ) del _pt_tmp_161 _pt_tmp_169 = actx.tag_axis(0, (DiscretizationElementAxisTag(),), _pt_tmp_169) del _pt_tmp_39, _pt_tmp_40 _pt_tmp_169 = actx.tag_axis(1, (DiscretizationDOFAxisTag(),), _pt_tmp_169) - _pt_tmp_168 = 0j + _pt_tmp_169 + _pt_tmp_168 = 0.0 + _pt_tmp_169 _pt_tmp_167 = ( _pt_tmp_168[_pt_tmp_41, _pt_tmp_42] if actx.permits_advanced_indexing else actx.call_loopy( - _pt_t_unit_4, in_0=_pt_tmp_168, in_1=_pt_tmp_41, in_2=_pt_tmp_42 + _pt_t_unit_2, in_0=_pt_tmp_168, in_1=_pt_tmp_41, in_2=_pt_tmp_42 )["out"] ) del _pt_tmp_169 _pt_tmp_167 = actx.tag_axis(0, (DiscretizationElementAxisTag(),), _pt_tmp_167) del _pt_tmp_41, _pt_tmp_42 _pt_tmp_167 = actx.tag_axis(1, (DiscretizationDOFAxisTag(),), _pt_tmp_167) - _pt_tmp_166 = 0j + _pt_tmp_167 + _pt_tmp_166 = 0.0 + _pt_tmp_167 _pt_tmp_165 = _pt_tmp_166 - _pt_tmp_168 del _pt_tmp_167 - _pt_tmp_164 = (1 + 0j) * _pt_tmp_165 + _pt_tmp_164 = 1.0 * _pt_tmp_165 del _pt_tmp_166, _pt_tmp_168 _pt_tmp_173 = ( _pt_tmp_31 * _pt_tmp_74 if actx.supports_nonscalar_broadcasting - else actx.call_loopy(_pt_t_unit_5, _in0=_pt_tmp_31, _in1=_pt_tmp_74)["out"] + else actx.call_loopy(_pt_t_unit_3, _in0=_pt_tmp_31, _in1=_pt_tmp_74)["out"] ) - _pt_tmp_172 = 0j + _pt_tmp_173 + _pt_tmp_172 = 0.0 + _pt_tmp_173 _pt_tmp_174 = ( _pt_tmp_61 * _pt_tmp_48 if actx.supports_nonscalar_broadcasting - else actx.call_loopy(_pt_t_unit_5, _in0=_pt_tmp_61, _in1=_pt_tmp_48)["out"] + else actx.call_loopy(_pt_t_unit_3, _in0=_pt_tmp_61, _in1=_pt_tmp_48)["out"] ) del _pt_tmp_173 _pt_tmp_171 = _pt_tmp_172 + _pt_tmp_174 - _pt_tmp_170 = (0.5 + 0j) * _pt_tmp_171 + _pt_tmp_170 = 0.5 * _pt_tmp_171 del _pt_tmp_172, _pt_tmp_174 _pt_tmp_163 = _pt_tmp_164 - _pt_tmp_170 del _pt_tmp_171 _pt_tmp_162 = ( _pt_tmp_80 * _pt_tmp_163 if actx.supports_nonscalar_broadcasting - else actx.call_loopy(_pt_t_unit_5, _in0=_pt_tmp_80, _in1=_pt_tmp_163)["out"] + else actx.call_loopy(_pt_t_unit_3, _in0=_pt_tmp_80, _in1=_pt_tmp_163)["out"] ) del _pt_tmp_164, _pt_tmp_170 _pt_tmp_159 = _pt_tmp_160 + _pt_tmp_162 - _pt_tmp_158 = (-0.5 + 0j) * _pt_tmp_159 + _pt_tmp_158 = -0.5 * _pt_tmp_159 del _pt_tmp_160, _pt_tmp_162 _pt_tmp_157 = ( _pt_tmp_158[_pt_tmp_81, _pt_tmp_82] if actx.permits_advanced_indexing else actx.call_loopy( - _pt_t_unit_7, in_0=_pt_tmp_158, in_1=_pt_tmp_81, in_2=_pt_tmp_82 + _pt_t_unit_5, in_0=_pt_tmp_158, in_1=_pt_tmp_81, in_2=_pt_tmp_82 )["out"] ) del _pt_tmp_159 _pt_tmp_156 = ( actx.np.where(_pt_tmp_25, _pt_tmp_157, 0) if actx.supports_nonscalar_broadcasting - else actx.call_loopy(_pt_t_unit_8, _in0=_pt_tmp_25, _in1=_pt_tmp_157)["out"] + else actx.call_loopy(_pt_t_unit_6, _in0=_pt_tmp_25, _in1=_pt_tmp_157)["out"] ) del _pt_tmp_158 _pt_tmp_156 = actx.tag_axis(0, (DiscretizationElementAxisTag(),), _pt_tmp_156) del _pt_tmp_157 _pt_tmp_156 = actx.tag_axis(1, (DiscretizationDOFAxisTag(),), _pt_tmp_156) - _pt_tmp_155 = 0j + _pt_tmp_156 - _pt_tmp_154 = 0j + _pt_tmp_155 + _pt_tmp_155 = 0.0 + _pt_tmp_156 + _pt_tmp_154 = 0.0 + _pt_tmp_155 del _pt_tmp_156 _pt_tmp_185 = ( _pt_tmp_129 * _pt_tmp_96 if actx.supports_nonscalar_broadcasting - else actx.call_loopy(_pt_t_unit_12, _in0=_pt_tmp_129, _in1=_pt_tmp_96)["out"] + else actx.call_loopy(_pt_t_unit_9, _in0=_pt_tmp_129, _in1=_pt_tmp_96)["out"] ) del _pt_tmp_155 - _pt_tmp_184 = 0j + _pt_tmp_185 + _pt_tmp_184 = 0.0 + _pt_tmp_185 del _pt_tmp_96 _pt_tmp_191 = ( _actx_in_1_3_0[_pt_tmp_101, _pt_tmp_102] if actx.permits_advanced_indexing else actx.call_loopy( - _pt_t_unit_11, in_0=_actx_in_1_3_0, in_1=_pt_tmp_101, in_2=_pt_tmp_102 + _pt_t_unit_8, in_0=_actx_in_1_3_0, in_1=_pt_tmp_101, in_2=_pt_tmp_102 )["out"] ) del _pt_tmp_185 _pt_tmp_191 = actx.tag_axis(0, (DiscretizationElementAxisTag(),), _pt_tmp_191) del _pt_tmp_101, _pt_tmp_102 _pt_tmp_191 = actx.tag_axis(1, (DiscretizationDOFAxisTag(),), _pt_tmp_191) - _pt_tmp_190 = 0j + _pt_tmp_191 + _pt_tmp_190 = 0.0 + _pt_tmp_191 _pt_tmp_189 = _pt_tmp_190 - _pt_tmp_190 del _pt_tmp_191 - _pt_tmp_188 = (1 + 0j) * _pt_tmp_189 + _pt_tmp_188 = 1.0 * _pt_tmp_189 del _pt_tmp_190 _pt_tmp_195 = ( _pt_tmp_95 * _pt_tmp_130 if actx.supports_nonscalar_broadcasting - else actx.call_loopy(_pt_t_unit_12, _in0=_pt_tmp_95, _in1=_pt_tmp_130)["out"] + else actx.call_loopy(_pt_t_unit_9, _in0=_pt_tmp_95, _in1=_pt_tmp_130)["out"] ) - _pt_tmp_194 = 0j + _pt_tmp_195 + _pt_tmp_194 = 0.0 + _pt_tmp_195 _pt_tmp_196 = ( _pt_tmp_119 * _pt_tmp_108 if actx.supports_nonscalar_broadcasting - else actx.call_loopy(_pt_t_unit_12, _in0=_pt_tmp_119, _in1=_pt_tmp_108)[ - "out" - ] + else actx.call_loopy(_pt_t_unit_9, _in0=_pt_tmp_119, _in1=_pt_tmp_108)["out"] ) del _pt_tmp_195 _pt_tmp_193 = _pt_tmp_194 + _pt_tmp_196 - _pt_tmp_192 = (0.5 + 0j) * _pt_tmp_193 + _pt_tmp_192 = 0.5 * _pt_tmp_193 del _pt_tmp_194, _pt_tmp_196 _pt_tmp_187 = _pt_tmp_188 - _pt_tmp_192 del _pt_tmp_193 _pt_tmp_186 = ( _pt_tmp_135 * _pt_tmp_187 if actx.supports_nonscalar_broadcasting - else actx.call_loopy(_pt_t_unit_12, _in0=_pt_tmp_135, _in1=_pt_tmp_187)[ - "out" - ] + else actx.call_loopy(_pt_t_unit_9, _in0=_pt_tmp_135, _in1=_pt_tmp_187)["out"] ) del _pt_tmp_188, _pt_tmp_192 _pt_tmp_183 = _pt_tmp_184 + _pt_tmp_186 - _pt_tmp_182 = (-0.5 + 0j) * _pt_tmp_183 + _pt_tmp_182 = -0.5 * _pt_tmp_183 del _pt_tmp_184, _pt_tmp_186 _pt_tmp_181 = ( _pt_tmp_182[_pt_tmp_136, _pt_tmp_82] if actx.permits_advanced_indexing else actx.call_loopy( - _pt_t_unit_13, in_0=_pt_tmp_182, in_1=_pt_tmp_136, in_2=_pt_tmp_82 + _pt_t_unit_10, in_0=_pt_tmp_182, in_1=_pt_tmp_136, in_2=_pt_tmp_82 )["out"] ) del _pt_tmp_183 _pt_tmp_180 = ( actx.np.where(_pt_tmp_89, _pt_tmp_181, 0) if actx.supports_nonscalar_broadcasting - else actx.call_loopy(_pt_t_unit_8, _in0=_pt_tmp_89, _in1=_pt_tmp_181)["out"] + else actx.call_loopy(_pt_t_unit_6, _in0=_pt_tmp_89, _in1=_pt_tmp_181)["out"] ) del _pt_tmp_182 _pt_tmp_180 = actx.tag_axis(0, (DiscretizationElementAxisTag(),), _pt_tmp_180) del _pt_tmp_181 _pt_tmp_180 = actx.tag_axis(1, (DiscretizationDOFAxisTag(),), _pt_tmp_180) - _pt_tmp_179 = 0j + _pt_tmp_180 - _pt_tmp_178 = 0j + _pt_tmp_179 + _pt_tmp_179 = 0.0 + _pt_tmp_180 + _pt_tmp_178 = 0.0 + _pt_tmp_179 del _pt_tmp_180 _pt_tmp_177 = _pt_tmp_178 + _pt_tmp_137 del _pt_tmp_179 @@ -849,105 +787,101 @@ def _rhs_inner( _pt_tmp_150 = actx.tag((FirstAxisIsElementsTag(),), _pt_tmp_150) del _pt_tmp_151 _pt_tmp_139 = _pt_tmp_140 - _pt_tmp_150 - _pt_tmp_138 = _pt_tmp_139 / np.complex128(1 + 0j) + _pt_tmp_138 = _pt_tmp_139 / np.float64(1.0) del _pt_tmp_140, _pt_tmp_150 _pt_tmp_204 = actx.einsum( "ij,ikl,jl->jk", _pt_tmp_146, _pt_data_0, _actx_in_1_4_0 ) del _pt_tmp_139 _pt_tmp_204 = actx.tag((FirstAxisIsElementsTag(),), _pt_tmp_204) - _pt_tmp_203 = (1 + 0j) * _pt_tmp_204 - _pt_tmp_202 = 0j + _pt_tmp_203 + _pt_tmp_203 = 1.0 * _pt_tmp_204 + _pt_tmp_202 = 0.0 + _pt_tmp_203 del _pt_tmp_204 _pt_tmp_206 = actx.einsum("ij,ikl,jl->jk", _pt_tmp_8, _pt_data_0, _actx_in_1_3_0) del _pt_tmp_203 _pt_tmp_206 = actx.tag((FirstAxisIsElementsTag(),), _pt_tmp_206) - _pt_tmp_205 = (-1 + 0j) * _pt_tmp_206 + _pt_tmp_205 = -1.0 * _pt_tmp_206 _pt_tmp_201 = _pt_tmp_202 + _pt_tmp_205 del _pt_tmp_206 - _pt_tmp_200 = 0j - _pt_tmp_201 + _pt_tmp_200 = 0.0 - _pt_tmp_201 del _pt_tmp_202, _pt_tmp_205 _pt_tmp_199 = -1 * _pt_tmp_200 del _pt_tmp_201 _pt_tmp_218 = ( _pt_tmp_47 * _pt_tmp_62 if actx.supports_nonscalar_broadcasting - else actx.call_loopy(_pt_t_unit_5, _in0=_pt_tmp_47, _in1=_pt_tmp_62)["out"] + else actx.call_loopy(_pt_t_unit_3, _in0=_pt_tmp_47, _in1=_pt_tmp_62)["out"] ) del _pt_tmp_200 - _pt_tmp_217 = 0j + _pt_tmp_218 + _pt_tmp_217 = 0.0 + _pt_tmp_218 del _pt_tmp_62 _pt_tmp_219 = ( _pt_tmp_54 * _pt_tmp_163 if actx.supports_nonscalar_broadcasting - else actx.call_loopy(_pt_t_unit_5, _in0=_pt_tmp_54, _in1=_pt_tmp_163)["out"] + else actx.call_loopy(_pt_t_unit_3, _in0=_pt_tmp_54, _in1=_pt_tmp_163)["out"] ) del _pt_tmp_218 _pt_tmp_216 = _pt_tmp_217 + _pt_tmp_219 del _pt_tmp_163 - _pt_tmp_215 = (-0.5 + 0j) * _pt_tmp_216 + _pt_tmp_215 = -0.5 * _pt_tmp_216 del _pt_tmp_217, _pt_tmp_219 _pt_tmp_214 = ( _pt_tmp_215[_pt_tmp_81, _pt_tmp_82] if actx.permits_advanced_indexing else actx.call_loopy( - _pt_t_unit_7, in_0=_pt_tmp_215, in_1=_pt_tmp_81, in_2=_pt_tmp_82 + _pt_t_unit_5, in_0=_pt_tmp_215, in_1=_pt_tmp_81, in_2=_pt_tmp_82 )["out"] ) del _pt_tmp_216 _pt_tmp_213 = ( actx.np.where(_pt_tmp_25, _pt_tmp_214, 0) if actx.supports_nonscalar_broadcasting - else actx.call_loopy(_pt_t_unit_8, _in0=_pt_tmp_25, _in1=_pt_tmp_214)["out"] + else actx.call_loopy(_pt_t_unit_6, _in0=_pt_tmp_25, _in1=_pt_tmp_214)["out"] ) del _pt_tmp_215 _pt_tmp_213 = actx.tag_axis(0, (DiscretizationElementAxisTag(),), _pt_tmp_213) del _pt_tmp_214 _pt_tmp_213 = actx.tag_axis(1, (DiscretizationDOFAxisTag(),), _pt_tmp_213) - _pt_tmp_212 = 0j + _pt_tmp_213 - _pt_tmp_211 = 0j + _pt_tmp_212 + _pt_tmp_212 = 0.0 + _pt_tmp_213 + _pt_tmp_211 = 0.0 + _pt_tmp_212 del _pt_tmp_213 _pt_tmp_230 = ( _pt_tmp_107 * _pt_tmp_120 if actx.supports_nonscalar_broadcasting - else actx.call_loopy(_pt_t_unit_12, _in0=_pt_tmp_107, _in1=_pt_tmp_120)[ - "out" - ] + else actx.call_loopy(_pt_t_unit_9, _in0=_pt_tmp_107, _in1=_pt_tmp_120)["out"] ) del _pt_tmp_212 - _pt_tmp_229 = 0j + _pt_tmp_230 + _pt_tmp_229 = 0.0 + _pt_tmp_230 del _pt_tmp_120 _pt_tmp_231 = ( _pt_tmp_113 * _pt_tmp_187 if actx.supports_nonscalar_broadcasting - else actx.call_loopy(_pt_t_unit_12, _in0=_pt_tmp_113, _in1=_pt_tmp_187)[ - "out" - ] + else actx.call_loopy(_pt_t_unit_9, _in0=_pt_tmp_113, _in1=_pt_tmp_187)["out"] ) del _pt_tmp_230 _pt_tmp_228 = _pt_tmp_229 + _pt_tmp_231 del _pt_tmp_187 - _pt_tmp_227 = (-0.5 + 0j) * _pt_tmp_228 + _pt_tmp_227 = -0.5 * _pt_tmp_228 del _pt_tmp_229, _pt_tmp_231 _pt_tmp_226 = ( _pt_tmp_227[_pt_tmp_136, _pt_tmp_82] if actx.permits_advanced_indexing else actx.call_loopy( - _pt_t_unit_13, in_0=_pt_tmp_227, in_1=_pt_tmp_136, in_2=_pt_tmp_82 + _pt_t_unit_10, in_0=_pt_tmp_227, in_1=_pt_tmp_136, in_2=_pt_tmp_82 )["out"] ) del _pt_tmp_228 _pt_tmp_225 = ( actx.np.where(_pt_tmp_89, _pt_tmp_226, 0) if actx.supports_nonscalar_broadcasting - else actx.call_loopy(_pt_t_unit_8, _in0=_pt_tmp_89, _in1=_pt_tmp_226)["out"] + else actx.call_loopy(_pt_t_unit_6, _in0=_pt_tmp_89, _in1=_pt_tmp_226)["out"] ) del _pt_tmp_227 _pt_tmp_225 = actx.tag_axis(0, (DiscretizationElementAxisTag(),), _pt_tmp_225) del _pt_tmp_226 _pt_tmp_225 = actx.tag_axis(1, (DiscretizationDOFAxisTag(),), _pt_tmp_225) - _pt_tmp_224 = 0j + _pt_tmp_225 - _pt_tmp_223 = 0j + _pt_tmp_224 + _pt_tmp_224 = 0.0 + _pt_tmp_225 + _pt_tmp_223 = 0.0 + _pt_tmp_224 del _pt_tmp_225 _pt_tmp_222 = _pt_tmp_223 + _pt_tmp_137 del _pt_tmp_224 @@ -968,185 +902,177 @@ def _rhs_inner( _pt_tmp_207 = actx.tag((FirstAxisIsElementsTag(),), _pt_tmp_207) del _pt_tmp_208 _pt_tmp_198 = _pt_tmp_199 - _pt_tmp_207 - _pt_tmp_197 = _pt_tmp_198 / np.complex128(1 + 0j) + _pt_tmp_197 = _pt_tmp_198 / np.float64(1.0) del _pt_tmp_199, _pt_tmp_207 _pt_tmp_238 = actx.einsum("ij,ikl,jl->jk", _pt_tmp_8, _pt_data_0, _actx_in_1_2_0) del _pt_tmp_198 _pt_tmp_238 = actx.tag((FirstAxisIsElementsTag(),), _pt_tmp_238) - _pt_tmp_237 = (1 + 0j) * _pt_tmp_238 - _pt_tmp_236 = 0j + _pt_tmp_237 + _pt_tmp_237 = 1.0 * _pt_tmp_238 + _pt_tmp_236 = 0.0 + _pt_tmp_237 del _pt_tmp_238 _pt_tmp_240 = actx.einsum( "ij,ikl,jl->jk", _pt_tmp_12, _pt_data_0, _actx_in_1_1_0 ) del _pt_tmp_237 _pt_tmp_240 = actx.tag((FirstAxisIsElementsTag(),), _pt_tmp_240) - _pt_tmp_239 = (-1 + 0j) * _pt_tmp_240 + _pt_tmp_239 = -1.0 * _pt_tmp_240 _pt_tmp_235 = _pt_tmp_236 + _pt_tmp_239 del _pt_tmp_240 _pt_tmp_234 = -1 * _pt_tmp_235 del _pt_tmp_236, _pt_tmp_239 - _pt_tmp_254 = (1 + 0j) * _pt_tmp_74 + _pt_tmp_254 = 1.0 * _pt_tmp_74 del _pt_tmp_235 _pt_tmp_258 = ( _pt_tmp_47 * _pt_tmp_64 if actx.supports_nonscalar_broadcasting - else actx.call_loopy(_pt_t_unit_5, _in0=_pt_tmp_47, _in1=_pt_tmp_64)["out"] + else actx.call_loopy(_pt_t_unit_3, _in0=_pt_tmp_47, _in1=_pt_tmp_64)["out"] ) del _pt_tmp_74 - _pt_tmp_257 = 0j + _pt_tmp_258 + _pt_tmp_257 = 0.0 + _pt_tmp_258 _pt_tmp_259 = ( _pt_tmp_54 * _pt_tmp_165 if actx.supports_nonscalar_broadcasting - else actx.call_loopy(_pt_t_unit_5, _in0=_pt_tmp_54, _in1=_pt_tmp_165)["out"] + else actx.call_loopy(_pt_t_unit_3, _in0=_pt_tmp_54, _in1=_pt_tmp_165)["out"] ) del _pt_tmp_258 _pt_tmp_256 = _pt_tmp_257 + _pt_tmp_259 - _pt_tmp_255 = (0.5 + 0j) * _pt_tmp_256 + _pt_tmp_255 = 0.5 * _pt_tmp_256 del _pt_tmp_257, _pt_tmp_259 _pt_tmp_253 = _pt_tmp_254 + _pt_tmp_255 del _pt_tmp_256 _pt_tmp_252 = ( _pt_tmp_31 * _pt_tmp_253 if actx.supports_nonscalar_broadcasting - else actx.call_loopy(_pt_t_unit_5, _in0=_pt_tmp_31, _in1=_pt_tmp_253)["out"] + else actx.call_loopy(_pt_t_unit_3, _in0=_pt_tmp_31, _in1=_pt_tmp_253)["out"] ) del _pt_tmp_254, _pt_tmp_255 - _pt_tmp_251 = 0j + _pt_tmp_252 - _pt_tmp_262 = (1 + 0j) * _pt_tmp_48 + _pt_tmp_251 = 0.0 + _pt_tmp_252 + _pt_tmp_262 = 1.0 * _pt_tmp_48 del _pt_tmp_252 _pt_tmp_266 = ( _pt_tmp_73 * _pt_tmp_34 if actx.supports_nonscalar_broadcasting - else actx.call_loopy(_pt_t_unit_14, _in0=_pt_tmp_73, _in1=_pt_tmp_34)["out"] + else actx.call_loopy(_pt_t_unit_3, _in0=_pt_tmp_73, _in1=_pt_tmp_34)["out"] ) del _pt_tmp_48 _pt_tmp_265 = 0.0 + _pt_tmp_266 _pt_tmp_267 = ( _pt_tmp_80 * _pt_tmp_165 if actx.supports_nonscalar_broadcasting - else actx.call_loopy(_pt_t_unit_5, _in0=_pt_tmp_80, _in1=_pt_tmp_165)["out"] + else actx.call_loopy(_pt_t_unit_3, _in0=_pt_tmp_80, _in1=_pt_tmp_165)["out"] ) del _pt_tmp_266 _pt_tmp_264 = _pt_tmp_265 + _pt_tmp_267 del _pt_tmp_165 - _pt_tmp_263 = (0.5 + 0j) * _pt_tmp_264 + _pt_tmp_263 = 0.5 * _pt_tmp_264 del _pt_tmp_265, _pt_tmp_267 _pt_tmp_261 = _pt_tmp_262 + _pt_tmp_263 del _pt_tmp_264 _pt_tmp_260 = ( _pt_tmp_61 * _pt_tmp_261 if actx.supports_nonscalar_broadcasting - else actx.call_loopy(_pt_t_unit_5, _in0=_pt_tmp_61, _in1=_pt_tmp_261)["out"] + else actx.call_loopy(_pt_t_unit_3, _in0=_pt_tmp_61, _in1=_pt_tmp_261)["out"] ) del _pt_tmp_262, _pt_tmp_263 _pt_tmp_250 = _pt_tmp_251 + _pt_tmp_260 - _pt_tmp_249 = (0.5 + 0j) * _pt_tmp_250 + _pt_tmp_249 = 0.5 * _pt_tmp_250 del _pt_tmp_251, _pt_tmp_260 _pt_tmp_248 = ( _pt_tmp_249[_pt_tmp_81, _pt_tmp_82] if actx.permits_advanced_indexing else actx.call_loopy( - _pt_t_unit_7, in_0=_pt_tmp_249, in_1=_pt_tmp_81, in_2=_pt_tmp_82 + _pt_t_unit_5, in_0=_pt_tmp_249, in_1=_pt_tmp_81, in_2=_pt_tmp_82 )["out"] ) del _pt_tmp_250 _pt_tmp_247 = ( actx.np.where(_pt_tmp_25, _pt_tmp_248, 0) if actx.supports_nonscalar_broadcasting - else actx.call_loopy(_pt_t_unit_8, _in0=_pt_tmp_25, _in1=_pt_tmp_248)["out"] + else actx.call_loopy(_pt_t_unit_6, _in0=_pt_tmp_25, _in1=_pt_tmp_248)["out"] ) del _pt_tmp_249 _pt_tmp_247 = actx.tag_axis(0, (DiscretizationElementAxisTag(),), _pt_tmp_247) del _pt_tmp_248 _pt_tmp_247 = actx.tag_axis(1, (DiscretizationDOFAxisTag(),), _pt_tmp_247) - _pt_tmp_246 = 0j + _pt_tmp_247 - _pt_tmp_245 = 0j + _pt_tmp_246 + _pt_tmp_246 = 0.0 + _pt_tmp_247 + _pt_tmp_245 = 0.0 + _pt_tmp_246 del _pt_tmp_247 - _pt_tmp_280 = (1 + 0j) * _pt_tmp_130 + _pt_tmp_280 = 1.0 * _pt_tmp_130 del _pt_tmp_246 _pt_tmp_284 = ( _pt_tmp_107 * _pt_tmp_122 if actx.supports_nonscalar_broadcasting - else actx.call_loopy(_pt_t_unit_12, _in0=_pt_tmp_107, _in1=_pt_tmp_122)[ - "out" - ] + else actx.call_loopy(_pt_t_unit_9, _in0=_pt_tmp_107, _in1=_pt_tmp_122)["out"] ) del _pt_tmp_130 - _pt_tmp_283 = 0j + _pt_tmp_284 + _pt_tmp_283 = 0.0 + _pt_tmp_284 _pt_tmp_285 = ( _pt_tmp_113 * _pt_tmp_189 if actx.supports_nonscalar_broadcasting - else actx.call_loopy(_pt_t_unit_12, _in0=_pt_tmp_113, _in1=_pt_tmp_189)[ - "out" - ] + else actx.call_loopy(_pt_t_unit_9, _in0=_pt_tmp_113, _in1=_pt_tmp_189)["out"] ) del _pt_tmp_284 _pt_tmp_282 = _pt_tmp_283 + _pt_tmp_285 - _pt_tmp_281 = (0.5 + 0j) * _pt_tmp_282 + _pt_tmp_281 = 0.5 * _pt_tmp_282 del _pt_tmp_283, _pt_tmp_285 _pt_tmp_279 = _pt_tmp_280 + _pt_tmp_281 del _pt_tmp_282 _pt_tmp_278 = ( _pt_tmp_95 * _pt_tmp_279 if actx.supports_nonscalar_broadcasting - else actx.call_loopy(_pt_t_unit_12, _in0=_pt_tmp_95, _in1=_pt_tmp_279)["out"] + else actx.call_loopy(_pt_t_unit_9, _in0=_pt_tmp_95, _in1=_pt_tmp_279)["out"] ) del _pt_tmp_280, _pt_tmp_281 - _pt_tmp_277 = 0j + _pt_tmp_278 - _pt_tmp_288 = (1 + 0j) * _pt_tmp_108 + _pt_tmp_277 = 0.0 + _pt_tmp_278 + _pt_tmp_288 = 1.0 * _pt_tmp_108 del _pt_tmp_278 _pt_tmp_292 = ( _pt_tmp_129 * _pt_tmp_98 if actx.supports_nonscalar_broadcasting - else actx.call_loopy(_pt_t_unit_15, _in0=_pt_tmp_129, _in1=_pt_tmp_98)["out"] + else actx.call_loopy(_pt_t_unit_9, _in0=_pt_tmp_129, _in1=_pt_tmp_98)["out"] ) del _pt_tmp_108 _pt_tmp_291 = 0.0 + _pt_tmp_292 _pt_tmp_293 = ( _pt_tmp_135 * _pt_tmp_189 if actx.supports_nonscalar_broadcasting - else actx.call_loopy(_pt_t_unit_12, _in0=_pt_tmp_135, _in1=_pt_tmp_189)[ - "out" - ] + else actx.call_loopy(_pt_t_unit_9, _in0=_pt_tmp_135, _in1=_pt_tmp_189)["out"] ) del _pt_tmp_292 _pt_tmp_290 = _pt_tmp_291 + _pt_tmp_293 del _pt_tmp_189 - _pt_tmp_289 = (0.5 + 0j) * _pt_tmp_290 + _pt_tmp_289 = 0.5 * _pt_tmp_290 del _pt_tmp_291, _pt_tmp_293 _pt_tmp_287 = _pt_tmp_288 + _pt_tmp_289 del _pt_tmp_290 _pt_tmp_286 = ( _pt_tmp_119 * _pt_tmp_287 if actx.supports_nonscalar_broadcasting - else actx.call_loopy(_pt_t_unit_12, _in0=_pt_tmp_119, _in1=_pt_tmp_287)[ - "out" - ] + else actx.call_loopy(_pt_t_unit_9, _in0=_pt_tmp_119, _in1=_pt_tmp_287)["out"] ) del _pt_tmp_288, _pt_tmp_289 _pt_tmp_276 = _pt_tmp_277 + _pt_tmp_286 - _pt_tmp_275 = (0.5 + 0j) * _pt_tmp_276 + _pt_tmp_275 = 0.5 * _pt_tmp_276 del _pt_tmp_277, _pt_tmp_286 _pt_tmp_274 = ( _pt_tmp_275[_pt_tmp_136, _pt_tmp_82] if actx.permits_advanced_indexing else actx.call_loopy( - _pt_t_unit_13, in_0=_pt_tmp_275, in_1=_pt_tmp_136, in_2=_pt_tmp_82 + _pt_t_unit_10, in_0=_pt_tmp_275, in_1=_pt_tmp_136, in_2=_pt_tmp_82 )["out"] ) del _pt_tmp_276 _pt_tmp_273 = ( actx.np.where(_pt_tmp_89, _pt_tmp_274, 0) if actx.supports_nonscalar_broadcasting - else actx.call_loopy(_pt_t_unit_8, _in0=_pt_tmp_89, _in1=_pt_tmp_274)["out"] + else actx.call_loopy(_pt_t_unit_6, _in0=_pt_tmp_89, _in1=_pt_tmp_274)["out"] ) del _pt_tmp_275 _pt_tmp_273 = actx.tag_axis(0, (DiscretizationElementAxisTag(),), _pt_tmp_273) del _pt_tmp_274 _pt_tmp_273 = actx.tag_axis(1, (DiscretizationDOFAxisTag(),), _pt_tmp_273) - _pt_tmp_272 = 0j + _pt_tmp_273 - _pt_tmp_271 = 0j + _pt_tmp_272 + _pt_tmp_272 = 0.0 + _pt_tmp_273 + _pt_tmp_271 = 0.0 + _pt_tmp_272 del _pt_tmp_273 _pt_tmp_270 = _pt_tmp_271 + _pt_tmp_137 del _pt_tmp_272 @@ -1167,15 +1093,15 @@ def _rhs_inner( _pt_tmp_241 = actx.tag((FirstAxisIsElementsTag(),), _pt_tmp_241) del _pt_tmp_242 _pt_tmp_233 = _pt_tmp_234 - _pt_tmp_241 - _pt_tmp_232 = _pt_tmp_233 / np.complex128(1 + 0j) + _pt_tmp_232 = _pt_tmp_233 / np.float64(1.0) del _pt_tmp_234, _pt_tmp_241 _pt_tmp_300 = actx.einsum( "ij,ikl,jl->jk", _pt_tmp_146, _pt_data_0, _actx_in_1_2_0 ) del _pt_tmp_233 _pt_tmp_300 = actx.tag((FirstAxisIsElementsTag(),), _pt_tmp_300) - _pt_tmp_299 = (-1 + 0j) * _pt_tmp_300 - _pt_tmp_298 = 0j + _pt_tmp_299 + _pt_tmp_299 = -1.0 * _pt_tmp_300 + _pt_tmp_298 = 0.0 + _pt_tmp_299 del _pt_tmp_300 _pt_tmp_302 = actx.einsum( "ij,ikl,jl->jk", _pt_tmp_12, _pt_data_0, _actx_in_1_0_0 @@ -1183,7 +1109,7 @@ def _rhs_inner( del _pt_tmp_299 _pt_tmp_302 = actx.tag((FirstAxisIsElementsTag(),), _pt_tmp_302) del _pt_tmp_12 - _pt_tmp_301 = (1 + 0j) * _pt_tmp_302 + _pt_tmp_301 = 1.0 * _pt_tmp_302 _pt_tmp_297 = _pt_tmp_298 + _pt_tmp_301 del _pt_tmp_302 _pt_tmp_296 = -1 * _pt_tmp_297 @@ -1191,17 +1117,17 @@ def _rhs_inner( _pt_tmp_314 = ( _pt_tmp_73 * _pt_tmp_253 if actx.supports_nonscalar_broadcasting - else actx.call_loopy(_pt_t_unit_5, _in0=_pt_tmp_73, _in1=_pt_tmp_253)["out"] + else actx.call_loopy(_pt_t_unit_3, _in0=_pt_tmp_73, _in1=_pt_tmp_253)["out"] ) del _pt_tmp_297 - _pt_tmp_313 = 0j + _pt_tmp_314 + _pt_tmp_313 = 0.0 + _pt_tmp_314 del _pt_tmp_253, _pt_tmp_73 - _pt_tmp_317 = (1 + 0j) * _pt_tmp_55 + _pt_tmp_317 = 1.0 * _pt_tmp_55 del _pt_tmp_314 _pt_tmp_321 = ( _pt_tmp_31 * _pt_tmp_34 if actx.supports_nonscalar_broadcasting - else actx.call_loopy(_pt_t_unit_14, _in0=_pt_tmp_31, _in1=_pt_tmp_34)["out"] + else actx.call_loopy(_pt_t_unit_3, _in0=_pt_tmp_31, _in1=_pt_tmp_34)["out"] ) del _pt_tmp_55 _pt_tmp_320 = 0.0 + _pt_tmp_321 @@ -1209,61 +1135,59 @@ def _rhs_inner( _pt_tmp_322 = ( _pt_tmp_61 * _pt_tmp_64 if actx.supports_nonscalar_broadcasting - else actx.call_loopy(_pt_t_unit_5, _in0=_pt_tmp_61, _in1=_pt_tmp_64)["out"] + else actx.call_loopy(_pt_t_unit_3, _in0=_pt_tmp_61, _in1=_pt_tmp_64)["out"] ) del _pt_tmp_321 _pt_tmp_319 = _pt_tmp_320 + _pt_tmp_322 del _pt_tmp_61, _pt_tmp_64 - _pt_tmp_318 = (0.5 + 0j) * _pt_tmp_319 + _pt_tmp_318 = 0.5 * _pt_tmp_319 del _pt_tmp_320, _pt_tmp_322 _pt_tmp_316 = _pt_tmp_317 + _pt_tmp_318 del _pt_tmp_319 _pt_tmp_315 = ( _pt_tmp_80 * _pt_tmp_316 if actx.supports_nonscalar_broadcasting - else actx.call_loopy(_pt_t_unit_5, _in0=_pt_tmp_80, _in1=_pt_tmp_316)["out"] + else actx.call_loopy(_pt_t_unit_3, _in0=_pt_tmp_80, _in1=_pt_tmp_316)["out"] ) del _pt_tmp_317, _pt_tmp_318 _pt_tmp_312 = _pt_tmp_313 + _pt_tmp_315 del _pt_tmp_80 - _pt_tmp_311 = (0.5 + 0j) * _pt_tmp_312 + _pt_tmp_311 = 0.5 * _pt_tmp_312 del _pt_tmp_313, _pt_tmp_315 _pt_tmp_310 = ( _pt_tmp_311[_pt_tmp_81, _pt_tmp_82] if actx.permits_advanced_indexing else actx.call_loopy( - _pt_t_unit_7, in_0=_pt_tmp_311, in_1=_pt_tmp_81, in_2=_pt_tmp_82 + _pt_t_unit_5, in_0=_pt_tmp_311, in_1=_pt_tmp_81, in_2=_pt_tmp_82 )["out"] ) del _pt_tmp_312 _pt_tmp_309 = ( actx.np.where(_pt_tmp_25, _pt_tmp_310, 0) if actx.supports_nonscalar_broadcasting - else actx.call_loopy(_pt_t_unit_8, _in0=_pt_tmp_25, _in1=_pt_tmp_310)["out"] + else actx.call_loopy(_pt_t_unit_6, _in0=_pt_tmp_25, _in1=_pt_tmp_310)["out"] ) del _pt_tmp_311 _pt_tmp_309 = actx.tag_axis(0, (DiscretizationElementAxisTag(),), _pt_tmp_309) del _pt_tmp_310 _pt_tmp_309 = actx.tag_axis(1, (DiscretizationDOFAxisTag(),), _pt_tmp_309) - _pt_tmp_308 = 0j + _pt_tmp_309 - _pt_tmp_307 = 0j + _pt_tmp_308 + _pt_tmp_308 = 0.0 + _pt_tmp_309 + _pt_tmp_307 = 0.0 + _pt_tmp_308 del _pt_tmp_309 _pt_tmp_333 = ( _pt_tmp_129 * _pt_tmp_279 if actx.supports_nonscalar_broadcasting - else actx.call_loopy(_pt_t_unit_12, _in0=_pt_tmp_129, _in1=_pt_tmp_279)[ - "out" - ] + else actx.call_loopy(_pt_t_unit_9, _in0=_pt_tmp_129, _in1=_pt_tmp_279)["out"] ) del _pt_tmp_308 - _pt_tmp_332 = 0j + _pt_tmp_333 + _pt_tmp_332 = 0.0 + _pt_tmp_333 del _pt_tmp_129, _pt_tmp_279 - _pt_tmp_336 = (1 + 0j) * _pt_tmp_114 + _pt_tmp_336 = 1.0 * _pt_tmp_114 del _pt_tmp_333 _pt_tmp_340 = ( _pt_tmp_95 * _pt_tmp_98 if actx.supports_nonscalar_broadcasting - else actx.call_loopy(_pt_t_unit_15, _in0=_pt_tmp_95, _in1=_pt_tmp_98)["out"] + else actx.call_loopy(_pt_t_unit_9, _in0=_pt_tmp_95, _in1=_pt_tmp_98)["out"] ) del _pt_tmp_114 _pt_tmp_339 = 0.0 + _pt_tmp_340 @@ -1271,48 +1195,44 @@ def _rhs_inner( _pt_tmp_341 = ( _pt_tmp_119 * _pt_tmp_122 if actx.supports_nonscalar_broadcasting - else actx.call_loopy(_pt_t_unit_12, _in0=_pt_tmp_119, _in1=_pt_tmp_122)[ - "out" - ] + else actx.call_loopy(_pt_t_unit_9, _in0=_pt_tmp_119, _in1=_pt_tmp_122)["out"] ) del _pt_tmp_340 _pt_tmp_338 = _pt_tmp_339 + _pt_tmp_341 del _pt_tmp_119, _pt_tmp_122 - _pt_tmp_337 = (0.5 + 0j) * _pt_tmp_338 + _pt_tmp_337 = 0.5 * _pt_tmp_338 del _pt_tmp_339, _pt_tmp_341 _pt_tmp_335 = _pt_tmp_336 + _pt_tmp_337 del _pt_tmp_338 _pt_tmp_334 = ( _pt_tmp_135 * _pt_tmp_335 if actx.supports_nonscalar_broadcasting - else actx.call_loopy(_pt_t_unit_12, _in0=_pt_tmp_135, _in1=_pt_tmp_335)[ - "out" - ] + else actx.call_loopy(_pt_t_unit_9, _in0=_pt_tmp_135, _in1=_pt_tmp_335)["out"] ) del _pt_tmp_336, _pt_tmp_337 _pt_tmp_331 = _pt_tmp_332 + _pt_tmp_334 del _pt_tmp_135 - _pt_tmp_330 = (0.5 + 0j) * _pt_tmp_331 + _pt_tmp_330 = 0.5 * _pt_tmp_331 del _pt_tmp_332, _pt_tmp_334 _pt_tmp_329 = ( _pt_tmp_330[_pt_tmp_136, _pt_tmp_82] if actx.permits_advanced_indexing else actx.call_loopy( - _pt_t_unit_13, in_0=_pt_tmp_330, in_1=_pt_tmp_136, in_2=_pt_tmp_82 + _pt_t_unit_10, in_0=_pt_tmp_330, in_1=_pt_tmp_136, in_2=_pt_tmp_82 )["out"] ) del _pt_tmp_331 _pt_tmp_328 = ( actx.np.where(_pt_tmp_89, _pt_tmp_329, 0) if actx.supports_nonscalar_broadcasting - else actx.call_loopy(_pt_t_unit_8, _in0=_pt_tmp_89, _in1=_pt_tmp_329)["out"] + else actx.call_loopy(_pt_t_unit_6, _in0=_pt_tmp_89, _in1=_pt_tmp_329)["out"] ) del _pt_tmp_330 _pt_tmp_328 = actx.tag_axis(0, (DiscretizationElementAxisTag(),), _pt_tmp_328) del _pt_tmp_329 _pt_tmp_328 = actx.tag_axis(1, (DiscretizationDOFAxisTag(),), _pt_tmp_328) - _pt_tmp_327 = 0j + _pt_tmp_328 - _pt_tmp_326 = 0j + _pt_tmp_327 + _pt_tmp_327 = 0.0 + _pt_tmp_328 + _pt_tmp_326 = 0.0 + _pt_tmp_327 del _pt_tmp_328 _pt_tmp_325 = _pt_tmp_326 + _pt_tmp_137 del _pt_tmp_327 @@ -1333,7 +1253,7 @@ def _rhs_inner( _pt_tmp_303 = actx.tag((FirstAxisIsElementsTag(),), _pt_tmp_303) del _pt_tmp_304 _pt_tmp_295 = _pt_tmp_296 - _pt_tmp_303 - _pt_tmp_294 = _pt_tmp_295 / np.complex128(1 + 0j) + _pt_tmp_294 = _pt_tmp_295 / np.float64(1.0) del _pt_tmp_296, _pt_tmp_303 _pt_tmp_348 = actx.einsum( "ij,ikl,jl->jk", _pt_tmp_146, _pt_data_0, _actx_in_1_1_0 @@ -1341,14 +1261,14 @@ def _rhs_inner( del _pt_tmp_295 _pt_tmp_348 = actx.tag((FirstAxisIsElementsTag(),), _pt_tmp_348) del _pt_tmp_146 - _pt_tmp_347 = (1 + 0j) * _pt_tmp_348 - _pt_tmp_346 = 0j + _pt_tmp_347 + _pt_tmp_347 = 1.0 * _pt_tmp_348 + _pt_tmp_346 = 0.0 + _pt_tmp_347 del _pt_tmp_348 _pt_tmp_350 = actx.einsum("ij,ikl,jl->jk", _pt_tmp_8, _pt_data_0, _actx_in_1_0_0) del _pt_tmp_347 _pt_tmp_350 = actx.tag((FirstAxisIsElementsTag(),), _pt_tmp_350) del _pt_tmp_8 - _pt_tmp_349 = (-1 + 0j) * _pt_tmp_350 + _pt_tmp_349 = -1.0 * _pt_tmp_350 _pt_tmp_345 = _pt_tmp_346 + _pt_tmp_349 del _pt_tmp_350 _pt_tmp_344 = -1 * _pt_tmp_345 @@ -1356,82 +1276,78 @@ def _rhs_inner( _pt_tmp_362 = ( _pt_tmp_47 * _pt_tmp_261 if actx.supports_nonscalar_broadcasting - else actx.call_loopy(_pt_t_unit_5, _in0=_pt_tmp_47, _in1=_pt_tmp_261)["out"] + else actx.call_loopy(_pt_t_unit_3, _in0=_pt_tmp_47, _in1=_pt_tmp_261)["out"] ) del _pt_tmp_345 - _pt_tmp_361 = 0j + _pt_tmp_362 + _pt_tmp_361 = 0.0 + _pt_tmp_362 del _pt_tmp_261, _pt_tmp_47 _pt_tmp_363 = ( _pt_tmp_54 * _pt_tmp_316 if actx.supports_nonscalar_broadcasting - else actx.call_loopy(_pt_t_unit_5, _in0=_pt_tmp_54, _in1=_pt_tmp_316)["out"] + else actx.call_loopy(_pt_t_unit_3, _in0=_pt_tmp_54, _in1=_pt_tmp_316)["out"] ) del _pt_tmp_362 _pt_tmp_360 = _pt_tmp_361 + _pt_tmp_363 del _pt_tmp_316, _pt_tmp_54 - _pt_tmp_359 = (0.5 + 0j) * _pt_tmp_360 + _pt_tmp_359 = 0.5 * _pt_tmp_360 del _pt_tmp_361, _pt_tmp_363 _pt_tmp_358 = ( _pt_tmp_359[_pt_tmp_81, _pt_tmp_82] if actx.permits_advanced_indexing else actx.call_loopy( - _pt_t_unit_7, in_0=_pt_tmp_359, in_1=_pt_tmp_81, in_2=_pt_tmp_82 + _pt_t_unit_5, in_0=_pt_tmp_359, in_1=_pt_tmp_81, in_2=_pt_tmp_82 )["out"] ) del _pt_tmp_360 _pt_tmp_357 = ( actx.np.where(_pt_tmp_25, _pt_tmp_358, 0) if actx.supports_nonscalar_broadcasting - else actx.call_loopy(_pt_t_unit_8, _in0=_pt_tmp_25, _in1=_pt_tmp_358)["out"] + else actx.call_loopy(_pt_t_unit_6, _in0=_pt_tmp_25, _in1=_pt_tmp_358)["out"] ) del _pt_tmp_359, _pt_tmp_81 _pt_tmp_357 = actx.tag_axis(0, (DiscretizationElementAxisTag(),), _pt_tmp_357) del _pt_tmp_25, _pt_tmp_358 _pt_tmp_357 = actx.tag_axis(1, (DiscretizationDOFAxisTag(),), _pt_tmp_357) - _pt_tmp_356 = 0j + _pt_tmp_357 - _pt_tmp_355 = 0j + _pt_tmp_356 + _pt_tmp_356 = 0.0 + _pt_tmp_357 + _pt_tmp_355 = 0.0 + _pt_tmp_356 del _pt_tmp_357 _pt_tmp_374 = ( _pt_tmp_107 * _pt_tmp_287 if actx.supports_nonscalar_broadcasting - else actx.call_loopy(_pt_t_unit_12, _in0=_pt_tmp_107, _in1=_pt_tmp_287)[ - "out" - ] + else actx.call_loopy(_pt_t_unit_9, _in0=_pt_tmp_107, _in1=_pt_tmp_287)["out"] ) del _pt_tmp_356 - _pt_tmp_373 = 0j + _pt_tmp_374 + _pt_tmp_373 = 0.0 + _pt_tmp_374 del _pt_tmp_107, _pt_tmp_287 _pt_tmp_375 = ( _pt_tmp_113 * _pt_tmp_335 if actx.supports_nonscalar_broadcasting - else actx.call_loopy(_pt_t_unit_12, _in0=_pt_tmp_113, _in1=_pt_tmp_335)[ - "out" - ] + else actx.call_loopy(_pt_t_unit_9, _in0=_pt_tmp_113, _in1=_pt_tmp_335)["out"] ) del _pt_tmp_374 _pt_tmp_372 = _pt_tmp_373 + _pt_tmp_375 del _pt_tmp_113, _pt_tmp_335 - _pt_tmp_371 = (0.5 + 0j) * _pt_tmp_372 + _pt_tmp_371 = 0.5 * _pt_tmp_372 del _pt_tmp_373, _pt_tmp_375 _pt_tmp_370 = ( _pt_tmp_371[_pt_tmp_136, _pt_tmp_82] if actx.permits_advanced_indexing else actx.call_loopy( - _pt_t_unit_13, in_0=_pt_tmp_371, in_1=_pt_tmp_136, in_2=_pt_tmp_82 + _pt_t_unit_10, in_0=_pt_tmp_371, in_1=_pt_tmp_136, in_2=_pt_tmp_82 )["out"] ) del _pt_tmp_372 _pt_tmp_369 = ( actx.np.where(_pt_tmp_89, _pt_tmp_370, 0) if actx.supports_nonscalar_broadcasting - else actx.call_loopy(_pt_t_unit_8, _in0=_pt_tmp_89, _in1=_pt_tmp_370)["out"] + else actx.call_loopy(_pt_t_unit_6, _in0=_pt_tmp_89, _in1=_pt_tmp_370)["out"] ) del _pt_tmp_136, _pt_tmp_371, _pt_tmp_82 _pt_tmp_369 = actx.tag_axis(0, (DiscretizationElementAxisTag(),), _pt_tmp_369) del _pt_tmp_370, _pt_tmp_89 _pt_tmp_369 = actx.tag_axis(1, (DiscretizationDOFAxisTag(),), _pt_tmp_369) - _pt_tmp_368 = 0j + _pt_tmp_369 - _pt_tmp_367 = 0j + _pt_tmp_368 + _pt_tmp_368 = 0.0 + _pt_tmp_369 + _pt_tmp_367 = 0.0 + _pt_tmp_368 del _pt_tmp_369 _pt_tmp_366 = _pt_tmp_367 + _pt_tmp_137 del _pt_tmp_368 @@ -1452,7 +1368,7 @@ def _rhs_inner( _pt_tmp_351 = actx.tag((FirstAxisIsElementsTag(),), _pt_tmp_351) del _pt_tmp_15, _pt_tmp_352 _pt_tmp_343 = _pt_tmp_344 - _pt_tmp_351 - _pt_tmp_342 = _pt_tmp_343 / np.complex128(1 + 0j) + _pt_tmp_342 = _pt_tmp_343 / np.float64(1.0) del _pt_tmp_344, _pt_tmp_351 _pt_tmp = make_obj_array( [_pt_tmp_0, _pt_tmp_138, _pt_tmp_197, _pt_tmp_232, _pt_tmp_294, _pt_tmp_342] diff --git a/actx_dgfem_suite/suite/tiny_maxwell_3D_P2/ref_input_args.pkl.dvc b/actx_dgfem_suite/suite/tiny_maxwell_3D_P2/ref_input_args.pkl.dvc index 196b568..2ae380a 100644 --- a/actx_dgfem_suite/suite/tiny_maxwell_3D_P2/ref_input_args.pkl.dvc +++ b/actx_dgfem_suite/suite/tiny_maxwell_3D_P2/ref_input_args.pkl.dvc @@ -1,5 +1,5 @@ outs: -- md5: 922a477916152d72fcee5b015800285b - size: 143281 +- md5: cc2cf1eca056956ca72151ef6f658afe + size: 78446 hash: md5 path: ref_input_args.pkl diff --git a/actx_dgfem_suite/suite/tiny_maxwell_3D_P2/ref_outputs.pkl.dvc b/actx_dgfem_suite/suite/tiny_maxwell_3D_P2/ref_outputs.pkl.dvc index 79ada2d..0e424b8 100644 --- a/actx_dgfem_suite/suite/tiny_maxwell_3D_P2/ref_outputs.pkl.dvc +++ b/actx_dgfem_suite/suite/tiny_maxwell_3D_P2/ref_outputs.pkl.dvc @@ -1,5 +1,5 @@ outs: -- md5: f1f83fbabedd376ee3f2ef1f83f7122b - size: 156180 +- md5: 97d7a8a41970cfa3fca37cb4c34e8aef + size: 78410 hash: md5 path: ref_outputs.pkl diff --git a/actx_dgfem_suite/suite/tiny_maxwell_3D_P3/literals.npz.dvc b/actx_dgfem_suite/suite/tiny_maxwell_3D_P3/literals.npz.dvc index 8f72fab..2de1934 100644 --- a/actx_dgfem_suite/suite/tiny_maxwell_3D_P3/literals.npz.dvc +++ b/actx_dgfem_suite/suite/tiny_maxwell_3D_P3/literals.npz.dvc @@ -1,5 +1,5 @@ outs: -- md5: c75a5856f5d5bc34ba85b2bff0de3463 - size: 79604 +- md5: 5ff1c229e81fcc0f42f53bdb754e834e + size: 73204 hash: md5 path: literals.npz diff --git a/actx_dgfem_suite/suite/tiny_maxwell_3D_P3/main.py b/actx_dgfem_suite/suite/tiny_maxwell_3D_P3/main.py index 3ffa964..9b084ee 100644 --- a/actx_dgfem_suite/suite/tiny_maxwell_3D_P3/main.py +++ b/actx_dgfem_suite/suite/tiny_maxwell_3D_P3/main.py @@ -58,62 +58,15 @@ def _rhs_inner( ], ) _pt_t_unit_10 = make_loopy_program( - "{ [_0, _1] : 0 <= _0 <= 107 and 0 <= _1 <= 9 }", - "out[_0, _1] = in_0[in_1[_0, 0] % 162, in_2[_0, _1] % 20]", - [ - lp.GlobalArg("out", dtype=np.float64, shape=(108, 10)), - lp.GlobalArg("in_0", dtype=np.float64, shape=(162, 20)), - lp.GlobalArg("in_1", dtype=np.int32, shape=(108, 1)), - lp.GlobalArg("in_2", dtype=np.int64, shape=(108, 10)), - ], - ) - _pt_t_unit_11 = make_loopy_program( - "{ [_0, _1] : 0 <= _0 <= 107 and 0 <= _1 <= 9 }", - "out[_0, _1] = in_0[in_1[_0, 0] % 162, in_2[_0, _1] % 20]", - [ - lp.GlobalArg("out", dtype=np.complex128, shape=(108, 10)), - lp.GlobalArg("in_0", dtype=np.complex128, shape=(162, 20)), - lp.GlobalArg("in_1", dtype=np.int32, shape=(108, 1)), - lp.GlobalArg("in_2", dtype=np.int64, shape=(108, 10)), - ], - ) - _pt_t_unit_12 = make_loopy_program( - "{ [_0, _1] : 0 <= _0 <= 107 and 0 <= _1 <= 9 }", - "out[_0, _1] = cast(np:dtype('complex128'), _in0[_0, 0])*_in1[_0, _1]", - [ - lp.GlobalArg("out", dtype=np.complex128, shape=(108, 10)), - lp.GlobalArg("_in0", dtype=np.float64, shape=(108, 1)), - lp.GlobalArg("_in1", dtype=np.complex128, shape=(108, 10)), - ], - ) - _pt_t_unit_13 = make_loopy_program( "{ [_0, _1] : 0 <= _0 <= 647 and 0 <= _1 <= 9 }", "out[_0, _1] = in_0[in_1[_0, 0] % 108, in_2[_0, _1] % 10]", [ - lp.GlobalArg("out", dtype=np.complex128, shape=(648, 10)), - lp.GlobalArg("in_0", dtype=np.complex128, shape=(108, 10)), + lp.GlobalArg("out", dtype=np.float64, shape=(648, 10)), + lp.GlobalArg("in_0", dtype=np.float64, shape=(108, 10)), lp.GlobalArg("in_1", dtype=np.int32, shape=(648, 1)), lp.GlobalArg("in_2", dtype=np.int64, shape=(648, 10)), ], ) - _pt_t_unit_14 = make_loopy_program( - "{ [_0, _1] : 0 <= _0 <= 539 and 0 <= _1 <= 9 }", - "out[_0, _1] = _in0[_0, 0]*_in1[_0, _1]", - [ - lp.GlobalArg("out", dtype=np.float64, shape=(540, 10)), - lp.GlobalArg("_in0", dtype=np.float64, shape=(540, 1)), - lp.GlobalArg("_in1", dtype=np.float64, shape=(540, 10)), - ], - ) - _pt_t_unit_15 = make_loopy_program( - "{ [_0, _1] : 0 <= _0 <= 107 and 0 <= _1 <= 9 }", - "out[_0, _1] = _in0[_0, 0]*_in1[_0, _1]", - [ - lp.GlobalArg("out", dtype=np.float64, shape=(108, 10)), - lp.GlobalArg("_in0", dtype=np.float64, shape=(108, 1)), - lp.GlobalArg("_in1", dtype=np.float64, shape=(108, 10)), - ], - ) _pt_t_unit_2 = make_loopy_program( "{ [_0, _1] : 0 <= _0 <= 539 and 0 <= _1 <= 9 }", "out[_0, _1] = in_0[in_1[_0, 0] % 540, in_2[_0, _1] % 10]", @@ -126,34 +79,14 @@ def _rhs_inner( ) _pt_t_unit_3 = make_loopy_program( "{ [_0, _1] : 0 <= _0 <= 539 and 0 <= _1 <= 9 }", - "out[_0, _1] = in_0[in_1[_0, 0] % 162, in_2[_0, _1] % 20]", - [ - lp.GlobalArg("out", dtype=np.complex128, shape=(540, 10)), - lp.GlobalArg("in_0", dtype=np.complex128, shape=(162, 20)), - lp.GlobalArg("in_1", dtype=np.int32, shape=(540, 1)), - lp.GlobalArg("in_2", dtype=np.int64, shape=(540, 10)), - ], - ) - _pt_t_unit_4 = make_loopy_program( - "{ [_0, _1] : 0 <= _0 <= 539 and 0 <= _1 <= 9 }", - "out[_0, _1] = in_0[in_1[_0, 0] % 540, in_2[_0, _1] % 10]", - [ - lp.GlobalArg("out", dtype=np.complex128, shape=(540, 10)), - lp.GlobalArg("in_0", dtype=np.complex128, shape=(540, 10)), - lp.GlobalArg("in_1", dtype=np.int32, shape=(540, 1)), - lp.GlobalArg("in_2", dtype=np.int64, shape=(540, 10)), - ], - ) - _pt_t_unit_5 = make_loopy_program( - "{ [_0, _1] : 0 <= _0 <= 539 and 0 <= _1 <= 9 }", - "out[_0, _1] = cast(np:dtype('complex128'), _in0[_0, 0])*_in1[_0, _1]", + "out[_0, _1] = _in0[_0, 0]*_in1[_0, _1]", [ - lp.GlobalArg("out", dtype=np.complex128, shape=(540, 10)), + lp.GlobalArg("out", dtype=np.float64, shape=(540, 10)), lp.GlobalArg("_in0", dtype=np.float64, shape=(540, 1)), - lp.GlobalArg("_in1", dtype=np.complex128, shape=(540, 10)), + lp.GlobalArg("_in1", dtype=np.float64, shape=(540, 10)), ], ) - _pt_t_unit_6 = make_loopy_program( + _pt_t_unit_4 = make_loopy_program( "{ [_0, _1] : 0 <= _0 <= 647 and 0 <= _1 <= 9 }", "out[_0, _1] = in_0[in_1[_0] % 1, _1]", [ @@ -162,26 +95,26 @@ def _rhs_inner( lp.GlobalArg("in_1", dtype=np.int8, shape=(648,)), ], ) - _pt_t_unit_7 = make_loopy_program( + _pt_t_unit_5 = make_loopy_program( "{ [_0, _1] : 0 <= _0 <= 647 and 0 <= _1 <= 9 }", "out[_0, _1] = in_0[in_1[_0, 0] % 540, in_2[_0, _1] % 10]", [ - lp.GlobalArg("out", dtype=np.complex128, shape=(648, 10)), - lp.GlobalArg("in_0", dtype=np.complex128, shape=(540, 10)), + lp.GlobalArg("out", dtype=np.float64, shape=(648, 10)), + lp.GlobalArg("in_0", dtype=np.float64, shape=(540, 10)), lp.GlobalArg("in_1", dtype=np.int32, shape=(648, 1)), lp.GlobalArg("in_2", dtype=np.int64, shape=(648, 10)), ], ) - _pt_t_unit_8 = make_loopy_program( + _pt_t_unit_6 = make_loopy_program( "{ [_0, _1] : 0 <= _0 <= 647 and 0 <= _1 <= 9 }", "out[_0, _1] = _in1[_0, _1] if _in0[_0, 0] else 0", [ - lp.GlobalArg("out", dtype=np.complex128, shape=(648, 10)), + lp.GlobalArg("out", dtype=np.float64, shape=(648, 10)), lp.GlobalArg("_in0", dtype=np.int8, shape=(648, 1)), - lp.GlobalArg("_in1", dtype=np.complex128, shape=(648, 10)), + lp.GlobalArg("_in1", dtype=np.float64, shape=(648, 10)), ], ) - _pt_t_unit_9 = make_loopy_program( + _pt_t_unit_7 = make_loopy_program( "{ [_0, _1] : 0 <= _0 <= 107 and 0 <= _1 <= 9 }", "out[_0, _1] = in_0[in_1[_0] % 4, _1]", [ @@ -190,6 +123,25 @@ def _rhs_inner( lp.GlobalArg("in_1", dtype=np.int8, shape=(108,)), ], ) + _pt_t_unit_8 = make_loopy_program( + "{ [_0, _1] : 0 <= _0 <= 107 and 0 <= _1 <= 9 }", + "out[_0, _1] = in_0[in_1[_0, 0] % 162, in_2[_0, _1] % 20]", + [ + lp.GlobalArg("out", dtype=np.float64, shape=(108, 10)), + lp.GlobalArg("in_0", dtype=np.float64, shape=(162, 20)), + lp.GlobalArg("in_1", dtype=np.int32, shape=(108, 1)), + lp.GlobalArg("in_2", dtype=np.int64, shape=(108, 10)), + ], + ) + _pt_t_unit_9 = make_loopy_program( + "{ [_0, _1] : 0 <= _0 <= 107 and 0 <= _1 <= 9 }", + "out[_0, _1] = _in0[_0, 0]*_in1[_0, _1]", + [ + lp.GlobalArg("out", dtype=np.float64, shape=(108, 10)), + lp.GlobalArg("_in0", dtype=np.float64, shape=(108, 1)), + lp.GlobalArg("_in1", dtype=np.float64, shape=(108, 10)), + ], + ) _pt_data = actx.thaw(npzfile["_pt_data"]) _pt_tmp_9 = _pt_data[1] _pt_tmp_8 = _pt_tmp_9[:, :, 0] @@ -206,10 +158,10 @@ def _rhs_inner( _pt_tmp_11 = actx.einsum("ij,ikl,jl->jk", _pt_tmp_12, _pt_data_0, _actx_in_1_4_0) del _pt_tmp_13 _pt_tmp_11 = actx.tag((FirstAxisIsElementsTag(),), _pt_tmp_11) - _pt_tmp_10 = (-1 + 0j) * _pt_tmp_11 + _pt_tmp_10 = -1.0 * _pt_tmp_11 _pt_tmp_4 = _pt_tmp_5 + _pt_tmp_10 del _pt_tmp_11 - _pt_tmp_3 = 0j - _pt_tmp_4 + _pt_tmp_3 = 0.0 - _pt_tmp_4 del _pt_tmp_10, _pt_tmp_5 _pt_tmp_2 = -1 * _pt_tmp_3 del _pt_tmp_4 @@ -278,75 +230,75 @@ def _rhs_inner( _actx_in_1_1_0[_pt_tmp_39, _pt_tmp_40] if actx.permits_advanced_indexing else actx.call_loopy( - _pt_t_unit_3, in_0=_actx_in_1_1_0, in_1=_pt_tmp_39, in_2=_pt_tmp_40 + _pt_t_unit_0, in_0=_actx_in_1_1_0, in_1=_pt_tmp_39, in_2=_pt_tmp_40 )["out"] ) _pt_tmp_52 = actx.tag_axis(0, (DiscretizationElementAxisTag(),), _pt_tmp_52) _pt_tmp_52 = actx.tag_axis(1, (DiscretizationDOFAxisTag(),), _pt_tmp_52) - _pt_tmp_51 = 0j + _pt_tmp_52 + _pt_tmp_51 = 0.0 + _pt_tmp_52 _pt_tmp_50 = ( _pt_tmp_51[_pt_tmp_41, _pt_tmp_42] if actx.permits_advanced_indexing else actx.call_loopy( - _pt_t_unit_4, in_0=_pt_tmp_51, in_1=_pt_tmp_41, in_2=_pt_tmp_42 + _pt_t_unit_2, in_0=_pt_tmp_51, in_1=_pt_tmp_41, in_2=_pt_tmp_42 )["out"] ) del _pt_tmp_52 _pt_tmp_50 = actx.tag_axis(0, (DiscretizationElementAxisTag(),), _pt_tmp_50) _pt_tmp_50 = actx.tag_axis(1, (DiscretizationDOFAxisTag(),), _pt_tmp_50) - _pt_tmp_49 = 0j + _pt_tmp_50 + _pt_tmp_49 = 0.0 + _pt_tmp_50 _pt_tmp_48 = _pt_tmp_49 - _pt_tmp_51 del _pt_tmp_50 _pt_tmp_46 = ( _pt_tmp_47 * _pt_tmp_48 if actx.supports_nonscalar_broadcasting - else actx.call_loopy(_pt_t_unit_5, _in0=_pt_tmp_47, _in1=_pt_tmp_48)["out"] + else actx.call_loopy(_pt_t_unit_3, _in0=_pt_tmp_47, _in1=_pt_tmp_48)["out"] ) del _pt_tmp_49, _pt_tmp_51 - _pt_tmp_45 = 0j + _pt_tmp_46 + _pt_tmp_45 = 0.0 + _pt_tmp_46 _pt_tmp_54 = -1.0 * _pt_data_6 del _pt_tmp_46 _pt_tmp_59 = ( _actx_in_1_0_0[_pt_tmp_39, _pt_tmp_40] if actx.permits_advanced_indexing else actx.call_loopy( - _pt_t_unit_3, in_0=_actx_in_1_0_0, in_1=_pt_tmp_39, in_2=_pt_tmp_40 + _pt_t_unit_0, in_0=_actx_in_1_0_0, in_1=_pt_tmp_39, in_2=_pt_tmp_40 )["out"] ) _pt_tmp_59 = actx.tag_axis(0, (DiscretizationElementAxisTag(),), _pt_tmp_59) _pt_tmp_59 = actx.tag_axis(1, (DiscretizationDOFAxisTag(),), _pt_tmp_59) - _pt_tmp_58 = 0j + _pt_tmp_59 + _pt_tmp_58 = 0.0 + _pt_tmp_59 _pt_tmp_57 = ( _pt_tmp_58[_pt_tmp_41, _pt_tmp_42] if actx.permits_advanced_indexing else actx.call_loopy( - _pt_t_unit_4, in_0=_pt_tmp_58, in_1=_pt_tmp_41, in_2=_pt_tmp_42 + _pt_t_unit_2, in_0=_pt_tmp_58, in_1=_pt_tmp_41, in_2=_pt_tmp_42 )["out"] ) del _pt_tmp_59 _pt_tmp_57 = actx.tag_axis(0, (DiscretizationElementAxisTag(),), _pt_tmp_57) _pt_tmp_57 = actx.tag_axis(1, (DiscretizationDOFAxisTag(),), _pt_tmp_57) - _pt_tmp_56 = 0j + _pt_tmp_57 + _pt_tmp_56 = 0.0 + _pt_tmp_57 _pt_tmp_55 = _pt_tmp_56 - _pt_tmp_58 del _pt_tmp_57 _pt_tmp_53 = ( _pt_tmp_54 * _pt_tmp_55 if actx.supports_nonscalar_broadcasting - else actx.call_loopy(_pt_t_unit_5, _in0=_pt_tmp_54, _in1=_pt_tmp_55)["out"] + else actx.call_loopy(_pt_t_unit_3, _in0=_pt_tmp_54, _in1=_pt_tmp_55)["out"] ) del _pt_tmp_56, _pt_tmp_58 _pt_tmp_44 = _pt_tmp_45 + _pt_tmp_53 - _pt_tmp_43 = (0.5 + 0j) * _pt_tmp_44 + _pt_tmp_43 = 0.5 * _pt_tmp_44 del _pt_tmp_45, _pt_tmp_53 _pt_tmp_32 = _pt_tmp_33 - _pt_tmp_43 del _pt_tmp_44 _pt_tmp_30 = ( _pt_tmp_31 * _pt_tmp_32 if actx.supports_nonscalar_broadcasting - else actx.call_loopy(_pt_t_unit_5, _in0=_pt_tmp_31, _in1=_pt_tmp_32)["out"] + else actx.call_loopy(_pt_t_unit_3, _in0=_pt_tmp_31, _in1=_pt_tmp_32)["out"] ) del _pt_tmp_33, _pt_tmp_43 - _pt_tmp_29 = 0j + _pt_tmp_30 + _pt_tmp_29 = 0.0 + _pt_tmp_30 _pt_data_14 = actx.thaw(npzfile["_pt_data_14"]) del _pt_tmp_30 _pt_tmp_61 = -1.0 * _pt_data_14 @@ -354,78 +306,78 @@ def _rhs_inner( _actx_in_1_4_0[_pt_tmp_39, _pt_tmp_40] if actx.permits_advanced_indexing else actx.call_loopy( - _pt_t_unit_3, in_0=_actx_in_1_4_0, in_1=_pt_tmp_39, in_2=_pt_tmp_40 + _pt_t_unit_0, in_0=_actx_in_1_4_0, in_1=_pt_tmp_39, in_2=_pt_tmp_40 )["out"] ) _pt_tmp_68 = actx.tag_axis(0, (DiscretizationElementAxisTag(),), _pt_tmp_68) _pt_tmp_68 = actx.tag_axis(1, (DiscretizationDOFAxisTag(),), _pt_tmp_68) - _pt_tmp_67 = 0j + _pt_tmp_68 + _pt_tmp_67 = 0.0 + _pt_tmp_68 _pt_tmp_66 = ( _pt_tmp_67[_pt_tmp_41, _pt_tmp_42] if actx.permits_advanced_indexing else actx.call_loopy( - _pt_t_unit_4, in_0=_pt_tmp_67, in_1=_pt_tmp_41, in_2=_pt_tmp_42 + _pt_t_unit_2, in_0=_pt_tmp_67, in_1=_pt_tmp_41, in_2=_pt_tmp_42 )["out"] ) del _pt_tmp_68 _pt_tmp_66 = actx.tag_axis(0, (DiscretizationElementAxisTag(),), _pt_tmp_66) _pt_tmp_66 = actx.tag_axis(1, (DiscretizationDOFAxisTag(),), _pt_tmp_66) - _pt_tmp_65 = 0j + _pt_tmp_66 + _pt_tmp_65 = 0.0 + _pt_tmp_66 _pt_tmp_64 = _pt_tmp_65 - _pt_tmp_67 del _pt_tmp_66 - _pt_tmp_63 = (1 + 0j) * _pt_tmp_64 + _pt_tmp_63 = 1.0 * _pt_tmp_64 del _pt_tmp_65, _pt_tmp_67 _pt_tmp_73 = -1.0 * _pt_data_13 _pt_tmp_78 = ( _actx_in_1_2_0[_pt_tmp_39, _pt_tmp_40] if actx.permits_advanced_indexing else actx.call_loopy( - _pt_t_unit_3, in_0=_actx_in_1_2_0, in_1=_pt_tmp_39, in_2=_pt_tmp_40 + _pt_t_unit_0, in_0=_actx_in_1_2_0, in_1=_pt_tmp_39, in_2=_pt_tmp_40 )["out"] ) _pt_tmp_78 = actx.tag_axis(0, (DiscretizationElementAxisTag(),), _pt_tmp_78) _pt_tmp_78 = actx.tag_axis(1, (DiscretizationDOFAxisTag(),), _pt_tmp_78) - _pt_tmp_77 = 0j + _pt_tmp_78 + _pt_tmp_77 = 0.0 + _pt_tmp_78 _pt_tmp_76 = ( _pt_tmp_77[_pt_tmp_41, _pt_tmp_42] if actx.permits_advanced_indexing else actx.call_loopy( - _pt_t_unit_4, in_0=_pt_tmp_77, in_1=_pt_tmp_41, in_2=_pt_tmp_42 + _pt_t_unit_2, in_0=_pt_tmp_77, in_1=_pt_tmp_41, in_2=_pt_tmp_42 )["out"] ) del _pt_tmp_78 _pt_tmp_76 = actx.tag_axis(0, (DiscretizationElementAxisTag(),), _pt_tmp_76) _pt_tmp_76 = actx.tag_axis(1, (DiscretizationDOFAxisTag(),), _pt_tmp_76) - _pt_tmp_75 = 0j + _pt_tmp_76 + _pt_tmp_75 = 0.0 + _pt_tmp_76 _pt_tmp_74 = _pt_tmp_75 - _pt_tmp_77 del _pt_tmp_76 _pt_tmp_72 = ( _pt_tmp_73 * _pt_tmp_74 if actx.supports_nonscalar_broadcasting - else actx.call_loopy(_pt_t_unit_5, _in0=_pt_tmp_73, _in1=_pt_tmp_74)["out"] + else actx.call_loopy(_pt_t_unit_3, _in0=_pt_tmp_73, _in1=_pt_tmp_74)["out"] ) del _pt_tmp_75, _pt_tmp_77 - _pt_tmp_71 = 0j + _pt_tmp_72 + _pt_tmp_71 = 0.0 + _pt_tmp_72 _pt_tmp_80 = 1.0 * _pt_data_14 del _pt_tmp_72 _pt_tmp_79 = ( _pt_tmp_80 * _pt_tmp_55 if actx.supports_nonscalar_broadcasting - else actx.call_loopy(_pt_t_unit_5, _in0=_pt_tmp_80, _in1=_pt_tmp_55)["out"] + else actx.call_loopy(_pt_t_unit_3, _in0=_pt_tmp_80, _in1=_pt_tmp_55)["out"] ) _pt_tmp_70 = _pt_tmp_71 + _pt_tmp_79 - _pt_tmp_69 = (0.5 + 0j) * _pt_tmp_70 + _pt_tmp_69 = 0.5 * _pt_tmp_70 del _pt_tmp_71, _pt_tmp_79 _pt_tmp_62 = _pt_tmp_63 - _pt_tmp_69 del _pt_tmp_70 _pt_tmp_60 = ( _pt_tmp_61 * _pt_tmp_62 if actx.supports_nonscalar_broadcasting - else actx.call_loopy(_pt_t_unit_5, _in0=_pt_tmp_61, _in1=_pt_tmp_62)["out"] + else actx.call_loopy(_pt_t_unit_3, _in0=_pt_tmp_61, _in1=_pt_tmp_62)["out"] ) del _pt_tmp_63, _pt_tmp_69 _pt_tmp_28 = _pt_tmp_29 + _pt_tmp_60 - _pt_tmp_27 = (-0.5 + 0j) * _pt_tmp_28 + _pt_tmp_27 = -0.5 * _pt_tmp_28 del _pt_tmp_29, _pt_tmp_60 _pt_data_15 = actx.thaw(npzfile["_pt_data_15"]) del _pt_tmp_28 @@ -435,26 +387,26 @@ def _rhs_inner( _pt_tmp_82 = ( _pt_data_16[_pt_data_17] if actx.permits_advanced_indexing - else actx.call_loopy(_pt_t_unit_6, in_0=_pt_data_16, in_1=_pt_data_17)["out"] + else actx.call_loopy(_pt_t_unit_4, in_0=_pt_data_16, in_1=_pt_data_17)["out"] ) _pt_tmp_26 = ( _pt_tmp_27[_pt_tmp_81, _pt_tmp_82] if actx.permits_advanced_indexing else actx.call_loopy( - _pt_t_unit_7, in_0=_pt_tmp_27, in_1=_pt_tmp_81, in_2=_pt_tmp_82 + _pt_t_unit_5, in_0=_pt_tmp_27, in_1=_pt_tmp_81, in_2=_pt_tmp_82 )["out"] ) _pt_tmp_24 = ( actx.np.where(_pt_tmp_25, _pt_tmp_26, 0) if actx.supports_nonscalar_broadcasting - else actx.call_loopy(_pt_t_unit_8, _in0=_pt_tmp_25, _in1=_pt_tmp_26)["out"] + else actx.call_loopy(_pt_t_unit_6, _in0=_pt_tmp_25, _in1=_pt_tmp_26)["out"] ) del _pt_tmp_27 _pt_tmp_24 = actx.tag_axis(0, (DiscretizationElementAxisTag(),), _pt_tmp_24) del _pt_tmp_26 _pt_tmp_24 = actx.tag_axis(1, (DiscretizationDOFAxisTag(),), _pt_tmp_24) - _pt_tmp_23 = 0j + _pt_tmp_24 - _pt_tmp_22 = 0j + _pt_tmp_23 + _pt_tmp_23 = 0.0 + _pt_tmp_24 + _pt_tmp_22 = 0.0 + _pt_tmp_23 del _pt_tmp_24 _pt_data_18 = actx.thaw(npzfile["_pt_data_18"]) del _pt_tmp_23 @@ -467,13 +419,13 @@ def _rhs_inner( _pt_tmp_102 = ( _pt_data_8[_pt_data_21] if actx.permits_advanced_indexing - else actx.call_loopy(_pt_t_unit_9, in_0=_pt_data_8, in_1=_pt_data_21)["out"] + else actx.call_loopy(_pt_t_unit_7, in_0=_pt_data_8, in_1=_pt_data_21)["out"] ) _pt_tmp_100 = ( _actx_in_1_5_0[_pt_tmp_101, _pt_tmp_102] if actx.permits_advanced_indexing else actx.call_loopy( - _pt_t_unit_10, in_0=_actx_in_1_5_0, in_1=_pt_tmp_101, in_2=_pt_tmp_102 + _pt_t_unit_8, in_0=_actx_in_1_5_0, in_1=_pt_tmp_101, in_2=_pt_tmp_102 )["out"] ) _pt_tmp_100 = actx.tag_axis(0, (DiscretizationElementAxisTag(),), _pt_tmp_100) @@ -489,59 +441,55 @@ def _rhs_inner( _actx_in_1_1_0[_pt_tmp_101, _pt_tmp_102] if actx.permits_advanced_indexing else actx.call_loopy( - _pt_t_unit_11, in_0=_actx_in_1_1_0, in_1=_pt_tmp_101, in_2=_pt_tmp_102 + _pt_t_unit_8, in_0=_actx_in_1_1_0, in_1=_pt_tmp_101, in_2=_pt_tmp_102 )["out"] ) _pt_tmp_111 = actx.tag_axis(0, (DiscretizationElementAxisTag(),), _pt_tmp_111) _pt_tmp_111 = actx.tag_axis(1, (DiscretizationDOFAxisTag(),), _pt_tmp_111) - _pt_tmp_110 = 0j + _pt_tmp_111 + _pt_tmp_110 = 0.0 + _pt_tmp_111 _pt_tmp_109 = -1 * _pt_tmp_110 del _pt_tmp_111 _pt_tmp_108 = _pt_tmp_109 - _pt_tmp_110 _pt_tmp_106 = ( _pt_tmp_107 * _pt_tmp_108 if actx.supports_nonscalar_broadcasting - else actx.call_loopy(_pt_t_unit_12, _in0=_pt_tmp_107, _in1=_pt_tmp_108)[ - "out" - ] + else actx.call_loopy(_pt_t_unit_9, _in0=_pt_tmp_107, _in1=_pt_tmp_108)["out"] ) del _pt_tmp_109, _pt_tmp_110 - _pt_tmp_105 = 0j + _pt_tmp_106 + _pt_tmp_105 = 0.0 + _pt_tmp_106 _pt_tmp_113 = -1.0 * _pt_data_19 del _pt_tmp_106 _pt_tmp_117 = ( _actx_in_1_0_0[_pt_tmp_101, _pt_tmp_102] if actx.permits_advanced_indexing else actx.call_loopy( - _pt_t_unit_11, in_0=_actx_in_1_0_0, in_1=_pt_tmp_101, in_2=_pt_tmp_102 + _pt_t_unit_8, in_0=_actx_in_1_0_0, in_1=_pt_tmp_101, in_2=_pt_tmp_102 )["out"] ) _pt_tmp_117 = actx.tag_axis(0, (DiscretizationElementAxisTag(),), _pt_tmp_117) _pt_tmp_117 = actx.tag_axis(1, (DiscretizationDOFAxisTag(),), _pt_tmp_117) - _pt_tmp_116 = 0j + _pt_tmp_117 + _pt_tmp_116 = 0.0 + _pt_tmp_117 _pt_tmp_115 = -1 * _pt_tmp_116 del _pt_tmp_117 _pt_tmp_114 = _pt_tmp_115 - _pt_tmp_116 _pt_tmp_112 = ( _pt_tmp_113 * _pt_tmp_114 if actx.supports_nonscalar_broadcasting - else actx.call_loopy(_pt_t_unit_12, _in0=_pt_tmp_113, _in1=_pt_tmp_114)[ - "out" - ] + else actx.call_loopy(_pt_t_unit_9, _in0=_pt_tmp_113, _in1=_pt_tmp_114)["out"] ) del _pt_tmp_115, _pt_tmp_116 _pt_tmp_104 = _pt_tmp_105 + _pt_tmp_112 - _pt_tmp_103 = (0.5 + 0j) * _pt_tmp_104 + _pt_tmp_103 = 0.5 * _pt_tmp_104 del _pt_tmp_105, _pt_tmp_112 _pt_tmp_96 = _pt_tmp_97 - _pt_tmp_103 del _pt_tmp_104 _pt_tmp_94 = ( _pt_tmp_95 * _pt_tmp_96 if actx.supports_nonscalar_broadcasting - else actx.call_loopy(_pt_t_unit_12, _in0=_pt_tmp_95, _in1=_pt_tmp_96)["out"] + else actx.call_loopy(_pt_t_unit_9, _in0=_pt_tmp_95, _in1=_pt_tmp_96)["out"] ) del _pt_tmp_103, _pt_tmp_97 - _pt_tmp_93 = 0j + _pt_tmp_94 + _pt_tmp_93 = 0.0 + _pt_tmp_94 _pt_data_23 = actx.thaw(npzfile["_pt_data_23"]) del _pt_tmp_94 _pt_tmp_119 = -1.0 * _pt_data_23 @@ -549,63 +497,57 @@ def _rhs_inner( _actx_in_1_4_0[_pt_tmp_101, _pt_tmp_102] if actx.permits_advanced_indexing else actx.call_loopy( - _pt_t_unit_11, in_0=_actx_in_1_4_0, in_1=_pt_tmp_101, in_2=_pt_tmp_102 + _pt_t_unit_8, in_0=_actx_in_1_4_0, in_1=_pt_tmp_101, in_2=_pt_tmp_102 )["out"] ) _pt_tmp_124 = actx.tag_axis(0, (DiscretizationElementAxisTag(),), _pt_tmp_124) _pt_tmp_124 = actx.tag_axis(1, (DiscretizationDOFAxisTag(),), _pt_tmp_124) - _pt_tmp_123 = 0j + _pt_tmp_124 + _pt_tmp_123 = 0.0 + _pt_tmp_124 _pt_tmp_122 = _pt_tmp_123 - _pt_tmp_123 del _pt_tmp_124 - _pt_tmp_121 = (1 + 0j) * _pt_tmp_122 + _pt_tmp_121 = 1.0 * _pt_tmp_122 del _pt_tmp_123 _pt_tmp_129 = -1.0 * _pt_data_22 _pt_tmp_133 = ( _actx_in_1_2_0[_pt_tmp_101, _pt_tmp_102] if actx.permits_advanced_indexing else actx.call_loopy( - _pt_t_unit_11, in_0=_actx_in_1_2_0, in_1=_pt_tmp_101, in_2=_pt_tmp_102 + _pt_t_unit_8, in_0=_actx_in_1_2_0, in_1=_pt_tmp_101, in_2=_pt_tmp_102 )["out"] ) _pt_tmp_133 = actx.tag_axis(0, (DiscretizationElementAxisTag(),), _pt_tmp_133) _pt_tmp_133 = actx.tag_axis(1, (DiscretizationDOFAxisTag(),), _pt_tmp_133) - _pt_tmp_132 = 0j + _pt_tmp_133 + _pt_tmp_132 = 0.0 + _pt_tmp_133 _pt_tmp_131 = -1 * _pt_tmp_132 del _pt_tmp_133 _pt_tmp_130 = _pt_tmp_131 - _pt_tmp_132 _pt_tmp_128 = ( _pt_tmp_129 * _pt_tmp_130 if actx.supports_nonscalar_broadcasting - else actx.call_loopy(_pt_t_unit_12, _in0=_pt_tmp_129, _in1=_pt_tmp_130)[ - "out" - ] + else actx.call_loopy(_pt_t_unit_9, _in0=_pt_tmp_129, _in1=_pt_tmp_130)["out"] ) del _pt_tmp_131, _pt_tmp_132 - _pt_tmp_127 = 0j + _pt_tmp_128 + _pt_tmp_127 = 0.0 + _pt_tmp_128 _pt_tmp_135 = 1.0 * _pt_data_23 del _pt_tmp_128 _pt_tmp_134 = ( _pt_tmp_135 * _pt_tmp_114 if actx.supports_nonscalar_broadcasting - else actx.call_loopy(_pt_t_unit_12, _in0=_pt_tmp_135, _in1=_pt_tmp_114)[ - "out" - ] + else actx.call_loopy(_pt_t_unit_9, _in0=_pt_tmp_135, _in1=_pt_tmp_114)["out"] ) _pt_tmp_126 = _pt_tmp_127 + _pt_tmp_134 - _pt_tmp_125 = (0.5 + 0j) * _pt_tmp_126 + _pt_tmp_125 = 0.5 * _pt_tmp_126 del _pt_tmp_127, _pt_tmp_134 _pt_tmp_120 = _pt_tmp_121 - _pt_tmp_125 del _pt_tmp_126 _pt_tmp_118 = ( _pt_tmp_119 * _pt_tmp_120 if actx.supports_nonscalar_broadcasting - else actx.call_loopy(_pt_t_unit_12, _in0=_pt_tmp_119, _in1=_pt_tmp_120)[ - "out" - ] + else actx.call_loopy(_pt_t_unit_9, _in0=_pt_tmp_119, _in1=_pt_tmp_120)["out"] ) del _pt_tmp_121, _pt_tmp_125 _pt_tmp_92 = _pt_tmp_93 + _pt_tmp_118 - _pt_tmp_91 = (-0.5 + 0j) * _pt_tmp_92 + _pt_tmp_91 = -0.5 * _pt_tmp_92 del _pt_tmp_118, _pt_tmp_93 _pt_data_24 = actx.thaw(npzfile["_pt_data_24"]) del _pt_tmp_92 @@ -614,22 +556,22 @@ def _rhs_inner( _pt_tmp_91[_pt_tmp_136, _pt_tmp_82] if actx.permits_advanced_indexing else actx.call_loopy( - _pt_t_unit_13, in_0=_pt_tmp_91, in_1=_pt_tmp_136, in_2=_pt_tmp_82 + _pt_t_unit_10, in_0=_pt_tmp_91, in_1=_pt_tmp_136, in_2=_pt_tmp_82 )["out"] ) _pt_tmp_88 = ( actx.np.where(_pt_tmp_89, _pt_tmp_90, 0) if actx.supports_nonscalar_broadcasting - else actx.call_loopy(_pt_t_unit_8, _in0=_pt_tmp_89, _in1=_pt_tmp_90)["out"] + else actx.call_loopy(_pt_t_unit_6, _in0=_pt_tmp_89, _in1=_pt_tmp_90)["out"] ) del _pt_tmp_91 _pt_tmp_88 = actx.tag_axis(0, (DiscretizationElementAxisTag(),), _pt_tmp_88) del _pt_tmp_90 _pt_tmp_88 = actx.tag_axis(1, (DiscretizationDOFAxisTag(),), _pt_tmp_88) - _pt_tmp_87 = 0j + _pt_tmp_88 - _pt_tmp_86 = 0j + _pt_tmp_87 + _pt_tmp_87 = 0.0 + _pt_tmp_88 + _pt_tmp_86 = 0.0 + _pt_tmp_87 del _pt_tmp_88 - _pt_tmp_137 = actx.np.zeros((648, 10), dtype=np.complex128) + _pt_tmp_137 = actx.np.zeros((648, 10), dtype=np.float64) del _pt_tmp_87 _pt_tmp_85 = _pt_tmp_86 + _pt_tmp_137 _pt_tmp_84 = _pt_tmp_85 + _pt_tmp_137 @@ -649,7 +591,7 @@ def _rhs_inner( _pt_tmp_14 = actx.tag((FirstAxisIsElementsTag(),), _pt_tmp_14) del _pt_tmp_17 _pt_tmp_1 = _pt_tmp_2 - _pt_tmp_14 - _pt_tmp_0 = _pt_tmp_1 / np.complex128(1 + 0j) + _pt_tmp_0 = _pt_tmp_1 / np.float64(1.0) del _pt_tmp_14, _pt_tmp_2 _pt_tmp_147 = _pt_data[0] del _pt_tmp_1 @@ -667,168 +609,164 @@ def _rhs_inner( ) del _pt_tmp_144 _pt_tmp_149 = actx.tag((FirstAxisIsElementsTag(),), _pt_tmp_149) - _pt_tmp_148 = (1 + 0j) * _pt_tmp_149 + _pt_tmp_148 = 1.0 * _pt_tmp_149 _pt_tmp_142 = _pt_tmp_143 + _pt_tmp_148 del _pt_tmp_149 - _pt_tmp_141 = 0j - _pt_tmp_142 + _pt_tmp_141 = 0.0 - _pt_tmp_142 del _pt_tmp_143, _pt_tmp_148 _pt_tmp_140 = -1 * _pt_tmp_141 del _pt_tmp_142 _pt_tmp_161 = ( _pt_tmp_73 * _pt_tmp_32 if actx.supports_nonscalar_broadcasting - else actx.call_loopy(_pt_t_unit_5, _in0=_pt_tmp_73, _in1=_pt_tmp_32)["out"] + else actx.call_loopy(_pt_t_unit_3, _in0=_pt_tmp_73, _in1=_pt_tmp_32)["out"] ) del _pt_tmp_141 - _pt_tmp_160 = 0j + _pt_tmp_161 + _pt_tmp_160 = 0.0 + _pt_tmp_161 del _pt_tmp_32 _pt_tmp_169 = ( _actx_in_1_3_0[_pt_tmp_39, _pt_tmp_40] if actx.permits_advanced_indexing else actx.call_loopy( - _pt_t_unit_3, in_0=_actx_in_1_3_0, in_1=_pt_tmp_39, in_2=_pt_tmp_40 + _pt_t_unit_0, in_0=_actx_in_1_3_0, in_1=_pt_tmp_39, in_2=_pt_tmp_40 )["out"] ) del _pt_tmp_161 _pt_tmp_169 = actx.tag_axis(0, (DiscretizationElementAxisTag(),), _pt_tmp_169) del _pt_tmp_39, _pt_tmp_40 _pt_tmp_169 = actx.tag_axis(1, (DiscretizationDOFAxisTag(),), _pt_tmp_169) - _pt_tmp_168 = 0j + _pt_tmp_169 + _pt_tmp_168 = 0.0 + _pt_tmp_169 _pt_tmp_167 = ( _pt_tmp_168[_pt_tmp_41, _pt_tmp_42] if actx.permits_advanced_indexing else actx.call_loopy( - _pt_t_unit_4, in_0=_pt_tmp_168, in_1=_pt_tmp_41, in_2=_pt_tmp_42 + _pt_t_unit_2, in_0=_pt_tmp_168, in_1=_pt_tmp_41, in_2=_pt_tmp_42 )["out"] ) del _pt_tmp_169 _pt_tmp_167 = actx.tag_axis(0, (DiscretizationElementAxisTag(),), _pt_tmp_167) del _pt_tmp_41, _pt_tmp_42 _pt_tmp_167 = actx.tag_axis(1, (DiscretizationDOFAxisTag(),), _pt_tmp_167) - _pt_tmp_166 = 0j + _pt_tmp_167 + _pt_tmp_166 = 0.0 + _pt_tmp_167 _pt_tmp_165 = _pt_tmp_166 - _pt_tmp_168 del _pt_tmp_167 - _pt_tmp_164 = (1 + 0j) * _pt_tmp_165 + _pt_tmp_164 = 1.0 * _pt_tmp_165 del _pt_tmp_166, _pt_tmp_168 _pt_tmp_173 = ( _pt_tmp_31 * _pt_tmp_74 if actx.supports_nonscalar_broadcasting - else actx.call_loopy(_pt_t_unit_5, _in0=_pt_tmp_31, _in1=_pt_tmp_74)["out"] + else actx.call_loopy(_pt_t_unit_3, _in0=_pt_tmp_31, _in1=_pt_tmp_74)["out"] ) - _pt_tmp_172 = 0j + _pt_tmp_173 + _pt_tmp_172 = 0.0 + _pt_tmp_173 _pt_tmp_174 = ( _pt_tmp_61 * _pt_tmp_48 if actx.supports_nonscalar_broadcasting - else actx.call_loopy(_pt_t_unit_5, _in0=_pt_tmp_61, _in1=_pt_tmp_48)["out"] + else actx.call_loopy(_pt_t_unit_3, _in0=_pt_tmp_61, _in1=_pt_tmp_48)["out"] ) del _pt_tmp_173 _pt_tmp_171 = _pt_tmp_172 + _pt_tmp_174 - _pt_tmp_170 = (0.5 + 0j) * _pt_tmp_171 + _pt_tmp_170 = 0.5 * _pt_tmp_171 del _pt_tmp_172, _pt_tmp_174 _pt_tmp_163 = _pt_tmp_164 - _pt_tmp_170 del _pt_tmp_171 _pt_tmp_162 = ( _pt_tmp_80 * _pt_tmp_163 if actx.supports_nonscalar_broadcasting - else actx.call_loopy(_pt_t_unit_5, _in0=_pt_tmp_80, _in1=_pt_tmp_163)["out"] + else actx.call_loopy(_pt_t_unit_3, _in0=_pt_tmp_80, _in1=_pt_tmp_163)["out"] ) del _pt_tmp_164, _pt_tmp_170 _pt_tmp_159 = _pt_tmp_160 + _pt_tmp_162 - _pt_tmp_158 = (-0.5 + 0j) * _pt_tmp_159 + _pt_tmp_158 = -0.5 * _pt_tmp_159 del _pt_tmp_160, _pt_tmp_162 _pt_tmp_157 = ( _pt_tmp_158[_pt_tmp_81, _pt_tmp_82] if actx.permits_advanced_indexing else actx.call_loopy( - _pt_t_unit_7, in_0=_pt_tmp_158, in_1=_pt_tmp_81, in_2=_pt_tmp_82 + _pt_t_unit_5, in_0=_pt_tmp_158, in_1=_pt_tmp_81, in_2=_pt_tmp_82 )["out"] ) del _pt_tmp_159 _pt_tmp_156 = ( actx.np.where(_pt_tmp_25, _pt_tmp_157, 0) if actx.supports_nonscalar_broadcasting - else actx.call_loopy(_pt_t_unit_8, _in0=_pt_tmp_25, _in1=_pt_tmp_157)["out"] + else actx.call_loopy(_pt_t_unit_6, _in0=_pt_tmp_25, _in1=_pt_tmp_157)["out"] ) del _pt_tmp_158 _pt_tmp_156 = actx.tag_axis(0, (DiscretizationElementAxisTag(),), _pt_tmp_156) del _pt_tmp_157 _pt_tmp_156 = actx.tag_axis(1, (DiscretizationDOFAxisTag(),), _pt_tmp_156) - _pt_tmp_155 = 0j + _pt_tmp_156 - _pt_tmp_154 = 0j + _pt_tmp_155 + _pt_tmp_155 = 0.0 + _pt_tmp_156 + _pt_tmp_154 = 0.0 + _pt_tmp_155 del _pt_tmp_156 _pt_tmp_185 = ( _pt_tmp_129 * _pt_tmp_96 if actx.supports_nonscalar_broadcasting - else actx.call_loopy(_pt_t_unit_12, _in0=_pt_tmp_129, _in1=_pt_tmp_96)["out"] + else actx.call_loopy(_pt_t_unit_9, _in0=_pt_tmp_129, _in1=_pt_tmp_96)["out"] ) del _pt_tmp_155 - _pt_tmp_184 = 0j + _pt_tmp_185 + _pt_tmp_184 = 0.0 + _pt_tmp_185 del _pt_tmp_96 _pt_tmp_191 = ( _actx_in_1_3_0[_pt_tmp_101, _pt_tmp_102] if actx.permits_advanced_indexing else actx.call_loopy( - _pt_t_unit_11, in_0=_actx_in_1_3_0, in_1=_pt_tmp_101, in_2=_pt_tmp_102 + _pt_t_unit_8, in_0=_actx_in_1_3_0, in_1=_pt_tmp_101, in_2=_pt_tmp_102 )["out"] ) del _pt_tmp_185 _pt_tmp_191 = actx.tag_axis(0, (DiscretizationElementAxisTag(),), _pt_tmp_191) del _pt_tmp_101, _pt_tmp_102 _pt_tmp_191 = actx.tag_axis(1, (DiscretizationDOFAxisTag(),), _pt_tmp_191) - _pt_tmp_190 = 0j + _pt_tmp_191 + _pt_tmp_190 = 0.0 + _pt_tmp_191 _pt_tmp_189 = _pt_tmp_190 - _pt_tmp_190 del _pt_tmp_191 - _pt_tmp_188 = (1 + 0j) * _pt_tmp_189 + _pt_tmp_188 = 1.0 * _pt_tmp_189 del _pt_tmp_190 _pt_tmp_195 = ( _pt_tmp_95 * _pt_tmp_130 if actx.supports_nonscalar_broadcasting - else actx.call_loopy(_pt_t_unit_12, _in0=_pt_tmp_95, _in1=_pt_tmp_130)["out"] + else actx.call_loopy(_pt_t_unit_9, _in0=_pt_tmp_95, _in1=_pt_tmp_130)["out"] ) - _pt_tmp_194 = 0j + _pt_tmp_195 + _pt_tmp_194 = 0.0 + _pt_tmp_195 _pt_tmp_196 = ( _pt_tmp_119 * _pt_tmp_108 if actx.supports_nonscalar_broadcasting - else actx.call_loopy(_pt_t_unit_12, _in0=_pt_tmp_119, _in1=_pt_tmp_108)[ - "out" - ] + else actx.call_loopy(_pt_t_unit_9, _in0=_pt_tmp_119, _in1=_pt_tmp_108)["out"] ) del _pt_tmp_195 _pt_tmp_193 = _pt_tmp_194 + _pt_tmp_196 - _pt_tmp_192 = (0.5 + 0j) * _pt_tmp_193 + _pt_tmp_192 = 0.5 * _pt_tmp_193 del _pt_tmp_194, _pt_tmp_196 _pt_tmp_187 = _pt_tmp_188 - _pt_tmp_192 del _pt_tmp_193 _pt_tmp_186 = ( _pt_tmp_135 * _pt_tmp_187 if actx.supports_nonscalar_broadcasting - else actx.call_loopy(_pt_t_unit_12, _in0=_pt_tmp_135, _in1=_pt_tmp_187)[ - "out" - ] + else actx.call_loopy(_pt_t_unit_9, _in0=_pt_tmp_135, _in1=_pt_tmp_187)["out"] ) del _pt_tmp_188, _pt_tmp_192 _pt_tmp_183 = _pt_tmp_184 + _pt_tmp_186 - _pt_tmp_182 = (-0.5 + 0j) * _pt_tmp_183 + _pt_tmp_182 = -0.5 * _pt_tmp_183 del _pt_tmp_184, _pt_tmp_186 _pt_tmp_181 = ( _pt_tmp_182[_pt_tmp_136, _pt_tmp_82] if actx.permits_advanced_indexing else actx.call_loopy( - _pt_t_unit_13, in_0=_pt_tmp_182, in_1=_pt_tmp_136, in_2=_pt_tmp_82 + _pt_t_unit_10, in_0=_pt_tmp_182, in_1=_pt_tmp_136, in_2=_pt_tmp_82 )["out"] ) del _pt_tmp_183 _pt_tmp_180 = ( actx.np.where(_pt_tmp_89, _pt_tmp_181, 0) if actx.supports_nonscalar_broadcasting - else actx.call_loopy(_pt_t_unit_8, _in0=_pt_tmp_89, _in1=_pt_tmp_181)["out"] + else actx.call_loopy(_pt_t_unit_6, _in0=_pt_tmp_89, _in1=_pt_tmp_181)["out"] ) del _pt_tmp_182 _pt_tmp_180 = actx.tag_axis(0, (DiscretizationElementAxisTag(),), _pt_tmp_180) del _pt_tmp_181 _pt_tmp_180 = actx.tag_axis(1, (DiscretizationDOFAxisTag(),), _pt_tmp_180) - _pt_tmp_179 = 0j + _pt_tmp_180 - _pt_tmp_178 = 0j + _pt_tmp_179 + _pt_tmp_179 = 0.0 + _pt_tmp_180 + _pt_tmp_178 = 0.0 + _pt_tmp_179 del _pt_tmp_180 _pt_tmp_177 = _pt_tmp_178 + _pt_tmp_137 del _pt_tmp_179 @@ -849,105 +787,101 @@ def _rhs_inner( _pt_tmp_150 = actx.tag((FirstAxisIsElementsTag(),), _pt_tmp_150) del _pt_tmp_151 _pt_tmp_139 = _pt_tmp_140 - _pt_tmp_150 - _pt_tmp_138 = _pt_tmp_139 / np.complex128(1 + 0j) + _pt_tmp_138 = _pt_tmp_139 / np.float64(1.0) del _pt_tmp_140, _pt_tmp_150 _pt_tmp_204 = actx.einsum( "ij,ikl,jl->jk", _pt_tmp_146, _pt_data_0, _actx_in_1_4_0 ) del _pt_tmp_139 _pt_tmp_204 = actx.tag((FirstAxisIsElementsTag(),), _pt_tmp_204) - _pt_tmp_203 = (1 + 0j) * _pt_tmp_204 - _pt_tmp_202 = 0j + _pt_tmp_203 + _pt_tmp_203 = 1.0 * _pt_tmp_204 + _pt_tmp_202 = 0.0 + _pt_tmp_203 del _pt_tmp_204 _pt_tmp_206 = actx.einsum("ij,ikl,jl->jk", _pt_tmp_8, _pt_data_0, _actx_in_1_3_0) del _pt_tmp_203 _pt_tmp_206 = actx.tag((FirstAxisIsElementsTag(),), _pt_tmp_206) - _pt_tmp_205 = (-1 + 0j) * _pt_tmp_206 + _pt_tmp_205 = -1.0 * _pt_tmp_206 _pt_tmp_201 = _pt_tmp_202 + _pt_tmp_205 del _pt_tmp_206 - _pt_tmp_200 = 0j - _pt_tmp_201 + _pt_tmp_200 = 0.0 - _pt_tmp_201 del _pt_tmp_202, _pt_tmp_205 _pt_tmp_199 = -1 * _pt_tmp_200 del _pt_tmp_201 _pt_tmp_218 = ( _pt_tmp_47 * _pt_tmp_62 if actx.supports_nonscalar_broadcasting - else actx.call_loopy(_pt_t_unit_5, _in0=_pt_tmp_47, _in1=_pt_tmp_62)["out"] + else actx.call_loopy(_pt_t_unit_3, _in0=_pt_tmp_47, _in1=_pt_tmp_62)["out"] ) del _pt_tmp_200 - _pt_tmp_217 = 0j + _pt_tmp_218 + _pt_tmp_217 = 0.0 + _pt_tmp_218 del _pt_tmp_62 _pt_tmp_219 = ( _pt_tmp_54 * _pt_tmp_163 if actx.supports_nonscalar_broadcasting - else actx.call_loopy(_pt_t_unit_5, _in0=_pt_tmp_54, _in1=_pt_tmp_163)["out"] + else actx.call_loopy(_pt_t_unit_3, _in0=_pt_tmp_54, _in1=_pt_tmp_163)["out"] ) del _pt_tmp_218 _pt_tmp_216 = _pt_tmp_217 + _pt_tmp_219 del _pt_tmp_163 - _pt_tmp_215 = (-0.5 + 0j) * _pt_tmp_216 + _pt_tmp_215 = -0.5 * _pt_tmp_216 del _pt_tmp_217, _pt_tmp_219 _pt_tmp_214 = ( _pt_tmp_215[_pt_tmp_81, _pt_tmp_82] if actx.permits_advanced_indexing else actx.call_loopy( - _pt_t_unit_7, in_0=_pt_tmp_215, in_1=_pt_tmp_81, in_2=_pt_tmp_82 + _pt_t_unit_5, in_0=_pt_tmp_215, in_1=_pt_tmp_81, in_2=_pt_tmp_82 )["out"] ) del _pt_tmp_216 _pt_tmp_213 = ( actx.np.where(_pt_tmp_25, _pt_tmp_214, 0) if actx.supports_nonscalar_broadcasting - else actx.call_loopy(_pt_t_unit_8, _in0=_pt_tmp_25, _in1=_pt_tmp_214)["out"] + else actx.call_loopy(_pt_t_unit_6, _in0=_pt_tmp_25, _in1=_pt_tmp_214)["out"] ) del _pt_tmp_215 _pt_tmp_213 = actx.tag_axis(0, (DiscretizationElementAxisTag(),), _pt_tmp_213) del _pt_tmp_214 _pt_tmp_213 = actx.tag_axis(1, (DiscretizationDOFAxisTag(),), _pt_tmp_213) - _pt_tmp_212 = 0j + _pt_tmp_213 - _pt_tmp_211 = 0j + _pt_tmp_212 + _pt_tmp_212 = 0.0 + _pt_tmp_213 + _pt_tmp_211 = 0.0 + _pt_tmp_212 del _pt_tmp_213 _pt_tmp_230 = ( _pt_tmp_107 * _pt_tmp_120 if actx.supports_nonscalar_broadcasting - else actx.call_loopy(_pt_t_unit_12, _in0=_pt_tmp_107, _in1=_pt_tmp_120)[ - "out" - ] + else actx.call_loopy(_pt_t_unit_9, _in0=_pt_tmp_107, _in1=_pt_tmp_120)["out"] ) del _pt_tmp_212 - _pt_tmp_229 = 0j + _pt_tmp_230 + _pt_tmp_229 = 0.0 + _pt_tmp_230 del _pt_tmp_120 _pt_tmp_231 = ( _pt_tmp_113 * _pt_tmp_187 if actx.supports_nonscalar_broadcasting - else actx.call_loopy(_pt_t_unit_12, _in0=_pt_tmp_113, _in1=_pt_tmp_187)[ - "out" - ] + else actx.call_loopy(_pt_t_unit_9, _in0=_pt_tmp_113, _in1=_pt_tmp_187)["out"] ) del _pt_tmp_230 _pt_tmp_228 = _pt_tmp_229 + _pt_tmp_231 del _pt_tmp_187 - _pt_tmp_227 = (-0.5 + 0j) * _pt_tmp_228 + _pt_tmp_227 = -0.5 * _pt_tmp_228 del _pt_tmp_229, _pt_tmp_231 _pt_tmp_226 = ( _pt_tmp_227[_pt_tmp_136, _pt_tmp_82] if actx.permits_advanced_indexing else actx.call_loopy( - _pt_t_unit_13, in_0=_pt_tmp_227, in_1=_pt_tmp_136, in_2=_pt_tmp_82 + _pt_t_unit_10, in_0=_pt_tmp_227, in_1=_pt_tmp_136, in_2=_pt_tmp_82 )["out"] ) del _pt_tmp_228 _pt_tmp_225 = ( actx.np.where(_pt_tmp_89, _pt_tmp_226, 0) if actx.supports_nonscalar_broadcasting - else actx.call_loopy(_pt_t_unit_8, _in0=_pt_tmp_89, _in1=_pt_tmp_226)["out"] + else actx.call_loopy(_pt_t_unit_6, _in0=_pt_tmp_89, _in1=_pt_tmp_226)["out"] ) del _pt_tmp_227 _pt_tmp_225 = actx.tag_axis(0, (DiscretizationElementAxisTag(),), _pt_tmp_225) del _pt_tmp_226 _pt_tmp_225 = actx.tag_axis(1, (DiscretizationDOFAxisTag(),), _pt_tmp_225) - _pt_tmp_224 = 0j + _pt_tmp_225 - _pt_tmp_223 = 0j + _pt_tmp_224 + _pt_tmp_224 = 0.0 + _pt_tmp_225 + _pt_tmp_223 = 0.0 + _pt_tmp_224 del _pt_tmp_225 _pt_tmp_222 = _pt_tmp_223 + _pt_tmp_137 del _pt_tmp_224 @@ -968,185 +902,177 @@ def _rhs_inner( _pt_tmp_207 = actx.tag((FirstAxisIsElementsTag(),), _pt_tmp_207) del _pt_tmp_208 _pt_tmp_198 = _pt_tmp_199 - _pt_tmp_207 - _pt_tmp_197 = _pt_tmp_198 / np.complex128(1 + 0j) + _pt_tmp_197 = _pt_tmp_198 / np.float64(1.0) del _pt_tmp_199, _pt_tmp_207 _pt_tmp_238 = actx.einsum("ij,ikl,jl->jk", _pt_tmp_8, _pt_data_0, _actx_in_1_2_0) del _pt_tmp_198 _pt_tmp_238 = actx.tag((FirstAxisIsElementsTag(),), _pt_tmp_238) - _pt_tmp_237 = (1 + 0j) * _pt_tmp_238 - _pt_tmp_236 = 0j + _pt_tmp_237 + _pt_tmp_237 = 1.0 * _pt_tmp_238 + _pt_tmp_236 = 0.0 + _pt_tmp_237 del _pt_tmp_238 _pt_tmp_240 = actx.einsum( "ij,ikl,jl->jk", _pt_tmp_12, _pt_data_0, _actx_in_1_1_0 ) del _pt_tmp_237 _pt_tmp_240 = actx.tag((FirstAxisIsElementsTag(),), _pt_tmp_240) - _pt_tmp_239 = (-1 + 0j) * _pt_tmp_240 + _pt_tmp_239 = -1.0 * _pt_tmp_240 _pt_tmp_235 = _pt_tmp_236 + _pt_tmp_239 del _pt_tmp_240 _pt_tmp_234 = -1 * _pt_tmp_235 del _pt_tmp_236, _pt_tmp_239 - _pt_tmp_254 = (1 + 0j) * _pt_tmp_74 + _pt_tmp_254 = 1.0 * _pt_tmp_74 del _pt_tmp_235 _pt_tmp_258 = ( _pt_tmp_47 * _pt_tmp_64 if actx.supports_nonscalar_broadcasting - else actx.call_loopy(_pt_t_unit_5, _in0=_pt_tmp_47, _in1=_pt_tmp_64)["out"] + else actx.call_loopy(_pt_t_unit_3, _in0=_pt_tmp_47, _in1=_pt_tmp_64)["out"] ) del _pt_tmp_74 - _pt_tmp_257 = 0j + _pt_tmp_258 + _pt_tmp_257 = 0.0 + _pt_tmp_258 _pt_tmp_259 = ( _pt_tmp_54 * _pt_tmp_165 if actx.supports_nonscalar_broadcasting - else actx.call_loopy(_pt_t_unit_5, _in0=_pt_tmp_54, _in1=_pt_tmp_165)["out"] + else actx.call_loopy(_pt_t_unit_3, _in0=_pt_tmp_54, _in1=_pt_tmp_165)["out"] ) del _pt_tmp_258 _pt_tmp_256 = _pt_tmp_257 + _pt_tmp_259 - _pt_tmp_255 = (0.5 + 0j) * _pt_tmp_256 + _pt_tmp_255 = 0.5 * _pt_tmp_256 del _pt_tmp_257, _pt_tmp_259 _pt_tmp_253 = _pt_tmp_254 + _pt_tmp_255 del _pt_tmp_256 _pt_tmp_252 = ( _pt_tmp_31 * _pt_tmp_253 if actx.supports_nonscalar_broadcasting - else actx.call_loopy(_pt_t_unit_5, _in0=_pt_tmp_31, _in1=_pt_tmp_253)["out"] + else actx.call_loopy(_pt_t_unit_3, _in0=_pt_tmp_31, _in1=_pt_tmp_253)["out"] ) del _pt_tmp_254, _pt_tmp_255 - _pt_tmp_251 = 0j + _pt_tmp_252 - _pt_tmp_262 = (1 + 0j) * _pt_tmp_48 + _pt_tmp_251 = 0.0 + _pt_tmp_252 + _pt_tmp_262 = 1.0 * _pt_tmp_48 del _pt_tmp_252 _pt_tmp_266 = ( _pt_tmp_73 * _pt_tmp_34 if actx.supports_nonscalar_broadcasting - else actx.call_loopy(_pt_t_unit_14, _in0=_pt_tmp_73, _in1=_pt_tmp_34)["out"] + else actx.call_loopy(_pt_t_unit_3, _in0=_pt_tmp_73, _in1=_pt_tmp_34)["out"] ) del _pt_tmp_48 _pt_tmp_265 = 0.0 + _pt_tmp_266 _pt_tmp_267 = ( _pt_tmp_80 * _pt_tmp_165 if actx.supports_nonscalar_broadcasting - else actx.call_loopy(_pt_t_unit_5, _in0=_pt_tmp_80, _in1=_pt_tmp_165)["out"] + else actx.call_loopy(_pt_t_unit_3, _in0=_pt_tmp_80, _in1=_pt_tmp_165)["out"] ) del _pt_tmp_266 _pt_tmp_264 = _pt_tmp_265 + _pt_tmp_267 del _pt_tmp_165 - _pt_tmp_263 = (0.5 + 0j) * _pt_tmp_264 + _pt_tmp_263 = 0.5 * _pt_tmp_264 del _pt_tmp_265, _pt_tmp_267 _pt_tmp_261 = _pt_tmp_262 + _pt_tmp_263 del _pt_tmp_264 _pt_tmp_260 = ( _pt_tmp_61 * _pt_tmp_261 if actx.supports_nonscalar_broadcasting - else actx.call_loopy(_pt_t_unit_5, _in0=_pt_tmp_61, _in1=_pt_tmp_261)["out"] + else actx.call_loopy(_pt_t_unit_3, _in0=_pt_tmp_61, _in1=_pt_tmp_261)["out"] ) del _pt_tmp_262, _pt_tmp_263 _pt_tmp_250 = _pt_tmp_251 + _pt_tmp_260 - _pt_tmp_249 = (0.5 + 0j) * _pt_tmp_250 + _pt_tmp_249 = 0.5 * _pt_tmp_250 del _pt_tmp_251, _pt_tmp_260 _pt_tmp_248 = ( _pt_tmp_249[_pt_tmp_81, _pt_tmp_82] if actx.permits_advanced_indexing else actx.call_loopy( - _pt_t_unit_7, in_0=_pt_tmp_249, in_1=_pt_tmp_81, in_2=_pt_tmp_82 + _pt_t_unit_5, in_0=_pt_tmp_249, in_1=_pt_tmp_81, in_2=_pt_tmp_82 )["out"] ) del _pt_tmp_250 _pt_tmp_247 = ( actx.np.where(_pt_tmp_25, _pt_tmp_248, 0) if actx.supports_nonscalar_broadcasting - else actx.call_loopy(_pt_t_unit_8, _in0=_pt_tmp_25, _in1=_pt_tmp_248)["out"] + else actx.call_loopy(_pt_t_unit_6, _in0=_pt_tmp_25, _in1=_pt_tmp_248)["out"] ) del _pt_tmp_249 _pt_tmp_247 = actx.tag_axis(0, (DiscretizationElementAxisTag(),), _pt_tmp_247) del _pt_tmp_248 _pt_tmp_247 = actx.tag_axis(1, (DiscretizationDOFAxisTag(),), _pt_tmp_247) - _pt_tmp_246 = 0j + _pt_tmp_247 - _pt_tmp_245 = 0j + _pt_tmp_246 + _pt_tmp_246 = 0.0 + _pt_tmp_247 + _pt_tmp_245 = 0.0 + _pt_tmp_246 del _pt_tmp_247 - _pt_tmp_280 = (1 + 0j) * _pt_tmp_130 + _pt_tmp_280 = 1.0 * _pt_tmp_130 del _pt_tmp_246 _pt_tmp_284 = ( _pt_tmp_107 * _pt_tmp_122 if actx.supports_nonscalar_broadcasting - else actx.call_loopy(_pt_t_unit_12, _in0=_pt_tmp_107, _in1=_pt_tmp_122)[ - "out" - ] + else actx.call_loopy(_pt_t_unit_9, _in0=_pt_tmp_107, _in1=_pt_tmp_122)["out"] ) del _pt_tmp_130 - _pt_tmp_283 = 0j + _pt_tmp_284 + _pt_tmp_283 = 0.0 + _pt_tmp_284 _pt_tmp_285 = ( _pt_tmp_113 * _pt_tmp_189 if actx.supports_nonscalar_broadcasting - else actx.call_loopy(_pt_t_unit_12, _in0=_pt_tmp_113, _in1=_pt_tmp_189)[ - "out" - ] + else actx.call_loopy(_pt_t_unit_9, _in0=_pt_tmp_113, _in1=_pt_tmp_189)["out"] ) del _pt_tmp_284 _pt_tmp_282 = _pt_tmp_283 + _pt_tmp_285 - _pt_tmp_281 = (0.5 + 0j) * _pt_tmp_282 + _pt_tmp_281 = 0.5 * _pt_tmp_282 del _pt_tmp_283, _pt_tmp_285 _pt_tmp_279 = _pt_tmp_280 + _pt_tmp_281 del _pt_tmp_282 _pt_tmp_278 = ( _pt_tmp_95 * _pt_tmp_279 if actx.supports_nonscalar_broadcasting - else actx.call_loopy(_pt_t_unit_12, _in0=_pt_tmp_95, _in1=_pt_tmp_279)["out"] + else actx.call_loopy(_pt_t_unit_9, _in0=_pt_tmp_95, _in1=_pt_tmp_279)["out"] ) del _pt_tmp_280, _pt_tmp_281 - _pt_tmp_277 = 0j + _pt_tmp_278 - _pt_tmp_288 = (1 + 0j) * _pt_tmp_108 + _pt_tmp_277 = 0.0 + _pt_tmp_278 + _pt_tmp_288 = 1.0 * _pt_tmp_108 del _pt_tmp_278 _pt_tmp_292 = ( _pt_tmp_129 * _pt_tmp_98 if actx.supports_nonscalar_broadcasting - else actx.call_loopy(_pt_t_unit_15, _in0=_pt_tmp_129, _in1=_pt_tmp_98)["out"] + else actx.call_loopy(_pt_t_unit_9, _in0=_pt_tmp_129, _in1=_pt_tmp_98)["out"] ) del _pt_tmp_108 _pt_tmp_291 = 0.0 + _pt_tmp_292 _pt_tmp_293 = ( _pt_tmp_135 * _pt_tmp_189 if actx.supports_nonscalar_broadcasting - else actx.call_loopy(_pt_t_unit_12, _in0=_pt_tmp_135, _in1=_pt_tmp_189)[ - "out" - ] + else actx.call_loopy(_pt_t_unit_9, _in0=_pt_tmp_135, _in1=_pt_tmp_189)["out"] ) del _pt_tmp_292 _pt_tmp_290 = _pt_tmp_291 + _pt_tmp_293 del _pt_tmp_189 - _pt_tmp_289 = (0.5 + 0j) * _pt_tmp_290 + _pt_tmp_289 = 0.5 * _pt_tmp_290 del _pt_tmp_291, _pt_tmp_293 _pt_tmp_287 = _pt_tmp_288 + _pt_tmp_289 del _pt_tmp_290 _pt_tmp_286 = ( _pt_tmp_119 * _pt_tmp_287 if actx.supports_nonscalar_broadcasting - else actx.call_loopy(_pt_t_unit_12, _in0=_pt_tmp_119, _in1=_pt_tmp_287)[ - "out" - ] + else actx.call_loopy(_pt_t_unit_9, _in0=_pt_tmp_119, _in1=_pt_tmp_287)["out"] ) del _pt_tmp_288, _pt_tmp_289 _pt_tmp_276 = _pt_tmp_277 + _pt_tmp_286 - _pt_tmp_275 = (0.5 + 0j) * _pt_tmp_276 + _pt_tmp_275 = 0.5 * _pt_tmp_276 del _pt_tmp_277, _pt_tmp_286 _pt_tmp_274 = ( _pt_tmp_275[_pt_tmp_136, _pt_tmp_82] if actx.permits_advanced_indexing else actx.call_loopy( - _pt_t_unit_13, in_0=_pt_tmp_275, in_1=_pt_tmp_136, in_2=_pt_tmp_82 + _pt_t_unit_10, in_0=_pt_tmp_275, in_1=_pt_tmp_136, in_2=_pt_tmp_82 )["out"] ) del _pt_tmp_276 _pt_tmp_273 = ( actx.np.where(_pt_tmp_89, _pt_tmp_274, 0) if actx.supports_nonscalar_broadcasting - else actx.call_loopy(_pt_t_unit_8, _in0=_pt_tmp_89, _in1=_pt_tmp_274)["out"] + else actx.call_loopy(_pt_t_unit_6, _in0=_pt_tmp_89, _in1=_pt_tmp_274)["out"] ) del _pt_tmp_275 _pt_tmp_273 = actx.tag_axis(0, (DiscretizationElementAxisTag(),), _pt_tmp_273) del _pt_tmp_274 _pt_tmp_273 = actx.tag_axis(1, (DiscretizationDOFAxisTag(),), _pt_tmp_273) - _pt_tmp_272 = 0j + _pt_tmp_273 - _pt_tmp_271 = 0j + _pt_tmp_272 + _pt_tmp_272 = 0.0 + _pt_tmp_273 + _pt_tmp_271 = 0.0 + _pt_tmp_272 del _pt_tmp_273 _pt_tmp_270 = _pt_tmp_271 + _pt_tmp_137 del _pt_tmp_272 @@ -1167,15 +1093,15 @@ def _rhs_inner( _pt_tmp_241 = actx.tag((FirstAxisIsElementsTag(),), _pt_tmp_241) del _pt_tmp_242 _pt_tmp_233 = _pt_tmp_234 - _pt_tmp_241 - _pt_tmp_232 = _pt_tmp_233 / np.complex128(1 + 0j) + _pt_tmp_232 = _pt_tmp_233 / np.float64(1.0) del _pt_tmp_234, _pt_tmp_241 _pt_tmp_300 = actx.einsum( "ij,ikl,jl->jk", _pt_tmp_146, _pt_data_0, _actx_in_1_2_0 ) del _pt_tmp_233 _pt_tmp_300 = actx.tag((FirstAxisIsElementsTag(),), _pt_tmp_300) - _pt_tmp_299 = (-1 + 0j) * _pt_tmp_300 - _pt_tmp_298 = 0j + _pt_tmp_299 + _pt_tmp_299 = -1.0 * _pt_tmp_300 + _pt_tmp_298 = 0.0 + _pt_tmp_299 del _pt_tmp_300 _pt_tmp_302 = actx.einsum( "ij,ikl,jl->jk", _pt_tmp_12, _pt_data_0, _actx_in_1_0_0 @@ -1183,7 +1109,7 @@ def _rhs_inner( del _pt_tmp_299 _pt_tmp_302 = actx.tag((FirstAxisIsElementsTag(),), _pt_tmp_302) del _pt_tmp_12 - _pt_tmp_301 = (1 + 0j) * _pt_tmp_302 + _pt_tmp_301 = 1.0 * _pt_tmp_302 _pt_tmp_297 = _pt_tmp_298 + _pt_tmp_301 del _pt_tmp_302 _pt_tmp_296 = -1 * _pt_tmp_297 @@ -1191,17 +1117,17 @@ def _rhs_inner( _pt_tmp_314 = ( _pt_tmp_73 * _pt_tmp_253 if actx.supports_nonscalar_broadcasting - else actx.call_loopy(_pt_t_unit_5, _in0=_pt_tmp_73, _in1=_pt_tmp_253)["out"] + else actx.call_loopy(_pt_t_unit_3, _in0=_pt_tmp_73, _in1=_pt_tmp_253)["out"] ) del _pt_tmp_297 - _pt_tmp_313 = 0j + _pt_tmp_314 + _pt_tmp_313 = 0.0 + _pt_tmp_314 del _pt_tmp_253, _pt_tmp_73 - _pt_tmp_317 = (1 + 0j) * _pt_tmp_55 + _pt_tmp_317 = 1.0 * _pt_tmp_55 del _pt_tmp_314 _pt_tmp_321 = ( _pt_tmp_31 * _pt_tmp_34 if actx.supports_nonscalar_broadcasting - else actx.call_loopy(_pt_t_unit_14, _in0=_pt_tmp_31, _in1=_pt_tmp_34)["out"] + else actx.call_loopy(_pt_t_unit_3, _in0=_pt_tmp_31, _in1=_pt_tmp_34)["out"] ) del _pt_tmp_55 _pt_tmp_320 = 0.0 + _pt_tmp_321 @@ -1209,61 +1135,59 @@ def _rhs_inner( _pt_tmp_322 = ( _pt_tmp_61 * _pt_tmp_64 if actx.supports_nonscalar_broadcasting - else actx.call_loopy(_pt_t_unit_5, _in0=_pt_tmp_61, _in1=_pt_tmp_64)["out"] + else actx.call_loopy(_pt_t_unit_3, _in0=_pt_tmp_61, _in1=_pt_tmp_64)["out"] ) del _pt_tmp_321 _pt_tmp_319 = _pt_tmp_320 + _pt_tmp_322 del _pt_tmp_61, _pt_tmp_64 - _pt_tmp_318 = (0.5 + 0j) * _pt_tmp_319 + _pt_tmp_318 = 0.5 * _pt_tmp_319 del _pt_tmp_320, _pt_tmp_322 _pt_tmp_316 = _pt_tmp_317 + _pt_tmp_318 del _pt_tmp_319 _pt_tmp_315 = ( _pt_tmp_80 * _pt_tmp_316 if actx.supports_nonscalar_broadcasting - else actx.call_loopy(_pt_t_unit_5, _in0=_pt_tmp_80, _in1=_pt_tmp_316)["out"] + else actx.call_loopy(_pt_t_unit_3, _in0=_pt_tmp_80, _in1=_pt_tmp_316)["out"] ) del _pt_tmp_317, _pt_tmp_318 _pt_tmp_312 = _pt_tmp_313 + _pt_tmp_315 del _pt_tmp_80 - _pt_tmp_311 = (0.5 + 0j) * _pt_tmp_312 + _pt_tmp_311 = 0.5 * _pt_tmp_312 del _pt_tmp_313, _pt_tmp_315 _pt_tmp_310 = ( _pt_tmp_311[_pt_tmp_81, _pt_tmp_82] if actx.permits_advanced_indexing else actx.call_loopy( - _pt_t_unit_7, in_0=_pt_tmp_311, in_1=_pt_tmp_81, in_2=_pt_tmp_82 + _pt_t_unit_5, in_0=_pt_tmp_311, in_1=_pt_tmp_81, in_2=_pt_tmp_82 )["out"] ) del _pt_tmp_312 _pt_tmp_309 = ( actx.np.where(_pt_tmp_25, _pt_tmp_310, 0) if actx.supports_nonscalar_broadcasting - else actx.call_loopy(_pt_t_unit_8, _in0=_pt_tmp_25, _in1=_pt_tmp_310)["out"] + else actx.call_loopy(_pt_t_unit_6, _in0=_pt_tmp_25, _in1=_pt_tmp_310)["out"] ) del _pt_tmp_311 _pt_tmp_309 = actx.tag_axis(0, (DiscretizationElementAxisTag(),), _pt_tmp_309) del _pt_tmp_310 _pt_tmp_309 = actx.tag_axis(1, (DiscretizationDOFAxisTag(),), _pt_tmp_309) - _pt_tmp_308 = 0j + _pt_tmp_309 - _pt_tmp_307 = 0j + _pt_tmp_308 + _pt_tmp_308 = 0.0 + _pt_tmp_309 + _pt_tmp_307 = 0.0 + _pt_tmp_308 del _pt_tmp_309 _pt_tmp_333 = ( _pt_tmp_129 * _pt_tmp_279 if actx.supports_nonscalar_broadcasting - else actx.call_loopy(_pt_t_unit_12, _in0=_pt_tmp_129, _in1=_pt_tmp_279)[ - "out" - ] + else actx.call_loopy(_pt_t_unit_9, _in0=_pt_tmp_129, _in1=_pt_tmp_279)["out"] ) del _pt_tmp_308 - _pt_tmp_332 = 0j + _pt_tmp_333 + _pt_tmp_332 = 0.0 + _pt_tmp_333 del _pt_tmp_129, _pt_tmp_279 - _pt_tmp_336 = (1 + 0j) * _pt_tmp_114 + _pt_tmp_336 = 1.0 * _pt_tmp_114 del _pt_tmp_333 _pt_tmp_340 = ( _pt_tmp_95 * _pt_tmp_98 if actx.supports_nonscalar_broadcasting - else actx.call_loopy(_pt_t_unit_15, _in0=_pt_tmp_95, _in1=_pt_tmp_98)["out"] + else actx.call_loopy(_pt_t_unit_9, _in0=_pt_tmp_95, _in1=_pt_tmp_98)["out"] ) del _pt_tmp_114 _pt_tmp_339 = 0.0 + _pt_tmp_340 @@ -1271,48 +1195,44 @@ def _rhs_inner( _pt_tmp_341 = ( _pt_tmp_119 * _pt_tmp_122 if actx.supports_nonscalar_broadcasting - else actx.call_loopy(_pt_t_unit_12, _in0=_pt_tmp_119, _in1=_pt_tmp_122)[ - "out" - ] + else actx.call_loopy(_pt_t_unit_9, _in0=_pt_tmp_119, _in1=_pt_tmp_122)["out"] ) del _pt_tmp_340 _pt_tmp_338 = _pt_tmp_339 + _pt_tmp_341 del _pt_tmp_119, _pt_tmp_122 - _pt_tmp_337 = (0.5 + 0j) * _pt_tmp_338 + _pt_tmp_337 = 0.5 * _pt_tmp_338 del _pt_tmp_339, _pt_tmp_341 _pt_tmp_335 = _pt_tmp_336 + _pt_tmp_337 del _pt_tmp_338 _pt_tmp_334 = ( _pt_tmp_135 * _pt_tmp_335 if actx.supports_nonscalar_broadcasting - else actx.call_loopy(_pt_t_unit_12, _in0=_pt_tmp_135, _in1=_pt_tmp_335)[ - "out" - ] + else actx.call_loopy(_pt_t_unit_9, _in0=_pt_tmp_135, _in1=_pt_tmp_335)["out"] ) del _pt_tmp_336, _pt_tmp_337 _pt_tmp_331 = _pt_tmp_332 + _pt_tmp_334 del _pt_tmp_135 - _pt_tmp_330 = (0.5 + 0j) * _pt_tmp_331 + _pt_tmp_330 = 0.5 * _pt_tmp_331 del _pt_tmp_332, _pt_tmp_334 _pt_tmp_329 = ( _pt_tmp_330[_pt_tmp_136, _pt_tmp_82] if actx.permits_advanced_indexing else actx.call_loopy( - _pt_t_unit_13, in_0=_pt_tmp_330, in_1=_pt_tmp_136, in_2=_pt_tmp_82 + _pt_t_unit_10, in_0=_pt_tmp_330, in_1=_pt_tmp_136, in_2=_pt_tmp_82 )["out"] ) del _pt_tmp_331 _pt_tmp_328 = ( actx.np.where(_pt_tmp_89, _pt_tmp_329, 0) if actx.supports_nonscalar_broadcasting - else actx.call_loopy(_pt_t_unit_8, _in0=_pt_tmp_89, _in1=_pt_tmp_329)["out"] + else actx.call_loopy(_pt_t_unit_6, _in0=_pt_tmp_89, _in1=_pt_tmp_329)["out"] ) del _pt_tmp_330 _pt_tmp_328 = actx.tag_axis(0, (DiscretizationElementAxisTag(),), _pt_tmp_328) del _pt_tmp_329 _pt_tmp_328 = actx.tag_axis(1, (DiscretizationDOFAxisTag(),), _pt_tmp_328) - _pt_tmp_327 = 0j + _pt_tmp_328 - _pt_tmp_326 = 0j + _pt_tmp_327 + _pt_tmp_327 = 0.0 + _pt_tmp_328 + _pt_tmp_326 = 0.0 + _pt_tmp_327 del _pt_tmp_328 _pt_tmp_325 = _pt_tmp_326 + _pt_tmp_137 del _pt_tmp_327 @@ -1333,7 +1253,7 @@ def _rhs_inner( _pt_tmp_303 = actx.tag((FirstAxisIsElementsTag(),), _pt_tmp_303) del _pt_tmp_304 _pt_tmp_295 = _pt_tmp_296 - _pt_tmp_303 - _pt_tmp_294 = _pt_tmp_295 / np.complex128(1 + 0j) + _pt_tmp_294 = _pt_tmp_295 / np.float64(1.0) del _pt_tmp_296, _pt_tmp_303 _pt_tmp_348 = actx.einsum( "ij,ikl,jl->jk", _pt_tmp_146, _pt_data_0, _actx_in_1_1_0 @@ -1341,14 +1261,14 @@ def _rhs_inner( del _pt_tmp_295 _pt_tmp_348 = actx.tag((FirstAxisIsElementsTag(),), _pt_tmp_348) del _pt_tmp_146 - _pt_tmp_347 = (1 + 0j) * _pt_tmp_348 - _pt_tmp_346 = 0j + _pt_tmp_347 + _pt_tmp_347 = 1.0 * _pt_tmp_348 + _pt_tmp_346 = 0.0 + _pt_tmp_347 del _pt_tmp_348 _pt_tmp_350 = actx.einsum("ij,ikl,jl->jk", _pt_tmp_8, _pt_data_0, _actx_in_1_0_0) del _pt_tmp_347 _pt_tmp_350 = actx.tag((FirstAxisIsElementsTag(),), _pt_tmp_350) del _pt_tmp_8 - _pt_tmp_349 = (-1 + 0j) * _pt_tmp_350 + _pt_tmp_349 = -1.0 * _pt_tmp_350 _pt_tmp_345 = _pt_tmp_346 + _pt_tmp_349 del _pt_tmp_350 _pt_tmp_344 = -1 * _pt_tmp_345 @@ -1356,82 +1276,78 @@ def _rhs_inner( _pt_tmp_362 = ( _pt_tmp_47 * _pt_tmp_261 if actx.supports_nonscalar_broadcasting - else actx.call_loopy(_pt_t_unit_5, _in0=_pt_tmp_47, _in1=_pt_tmp_261)["out"] + else actx.call_loopy(_pt_t_unit_3, _in0=_pt_tmp_47, _in1=_pt_tmp_261)["out"] ) del _pt_tmp_345 - _pt_tmp_361 = 0j + _pt_tmp_362 + _pt_tmp_361 = 0.0 + _pt_tmp_362 del _pt_tmp_261, _pt_tmp_47 _pt_tmp_363 = ( _pt_tmp_54 * _pt_tmp_316 if actx.supports_nonscalar_broadcasting - else actx.call_loopy(_pt_t_unit_5, _in0=_pt_tmp_54, _in1=_pt_tmp_316)["out"] + else actx.call_loopy(_pt_t_unit_3, _in0=_pt_tmp_54, _in1=_pt_tmp_316)["out"] ) del _pt_tmp_362 _pt_tmp_360 = _pt_tmp_361 + _pt_tmp_363 del _pt_tmp_316, _pt_tmp_54 - _pt_tmp_359 = (0.5 + 0j) * _pt_tmp_360 + _pt_tmp_359 = 0.5 * _pt_tmp_360 del _pt_tmp_361, _pt_tmp_363 _pt_tmp_358 = ( _pt_tmp_359[_pt_tmp_81, _pt_tmp_82] if actx.permits_advanced_indexing else actx.call_loopy( - _pt_t_unit_7, in_0=_pt_tmp_359, in_1=_pt_tmp_81, in_2=_pt_tmp_82 + _pt_t_unit_5, in_0=_pt_tmp_359, in_1=_pt_tmp_81, in_2=_pt_tmp_82 )["out"] ) del _pt_tmp_360 _pt_tmp_357 = ( actx.np.where(_pt_tmp_25, _pt_tmp_358, 0) if actx.supports_nonscalar_broadcasting - else actx.call_loopy(_pt_t_unit_8, _in0=_pt_tmp_25, _in1=_pt_tmp_358)["out"] + else actx.call_loopy(_pt_t_unit_6, _in0=_pt_tmp_25, _in1=_pt_tmp_358)["out"] ) del _pt_tmp_359, _pt_tmp_81 _pt_tmp_357 = actx.tag_axis(0, (DiscretizationElementAxisTag(),), _pt_tmp_357) del _pt_tmp_25, _pt_tmp_358 _pt_tmp_357 = actx.tag_axis(1, (DiscretizationDOFAxisTag(),), _pt_tmp_357) - _pt_tmp_356 = 0j + _pt_tmp_357 - _pt_tmp_355 = 0j + _pt_tmp_356 + _pt_tmp_356 = 0.0 + _pt_tmp_357 + _pt_tmp_355 = 0.0 + _pt_tmp_356 del _pt_tmp_357 _pt_tmp_374 = ( _pt_tmp_107 * _pt_tmp_287 if actx.supports_nonscalar_broadcasting - else actx.call_loopy(_pt_t_unit_12, _in0=_pt_tmp_107, _in1=_pt_tmp_287)[ - "out" - ] + else actx.call_loopy(_pt_t_unit_9, _in0=_pt_tmp_107, _in1=_pt_tmp_287)["out"] ) del _pt_tmp_356 - _pt_tmp_373 = 0j + _pt_tmp_374 + _pt_tmp_373 = 0.0 + _pt_tmp_374 del _pt_tmp_107, _pt_tmp_287 _pt_tmp_375 = ( _pt_tmp_113 * _pt_tmp_335 if actx.supports_nonscalar_broadcasting - else actx.call_loopy(_pt_t_unit_12, _in0=_pt_tmp_113, _in1=_pt_tmp_335)[ - "out" - ] + else actx.call_loopy(_pt_t_unit_9, _in0=_pt_tmp_113, _in1=_pt_tmp_335)["out"] ) del _pt_tmp_374 _pt_tmp_372 = _pt_tmp_373 + _pt_tmp_375 del _pt_tmp_113, _pt_tmp_335 - _pt_tmp_371 = (0.5 + 0j) * _pt_tmp_372 + _pt_tmp_371 = 0.5 * _pt_tmp_372 del _pt_tmp_373, _pt_tmp_375 _pt_tmp_370 = ( _pt_tmp_371[_pt_tmp_136, _pt_tmp_82] if actx.permits_advanced_indexing else actx.call_loopy( - _pt_t_unit_13, in_0=_pt_tmp_371, in_1=_pt_tmp_136, in_2=_pt_tmp_82 + _pt_t_unit_10, in_0=_pt_tmp_371, in_1=_pt_tmp_136, in_2=_pt_tmp_82 )["out"] ) del _pt_tmp_372 _pt_tmp_369 = ( actx.np.where(_pt_tmp_89, _pt_tmp_370, 0) if actx.supports_nonscalar_broadcasting - else actx.call_loopy(_pt_t_unit_8, _in0=_pt_tmp_89, _in1=_pt_tmp_370)["out"] + else actx.call_loopy(_pt_t_unit_6, _in0=_pt_tmp_89, _in1=_pt_tmp_370)["out"] ) del _pt_tmp_136, _pt_tmp_371, _pt_tmp_82 _pt_tmp_369 = actx.tag_axis(0, (DiscretizationElementAxisTag(),), _pt_tmp_369) del _pt_tmp_370, _pt_tmp_89 _pt_tmp_369 = actx.tag_axis(1, (DiscretizationDOFAxisTag(),), _pt_tmp_369) - _pt_tmp_368 = 0j + _pt_tmp_369 - _pt_tmp_367 = 0j + _pt_tmp_368 + _pt_tmp_368 = 0.0 + _pt_tmp_369 + _pt_tmp_367 = 0.0 + _pt_tmp_368 del _pt_tmp_369 _pt_tmp_366 = _pt_tmp_367 + _pt_tmp_137 del _pt_tmp_368 @@ -1452,7 +1368,7 @@ def _rhs_inner( _pt_tmp_351 = actx.tag((FirstAxisIsElementsTag(),), _pt_tmp_351) del _pt_tmp_15, _pt_tmp_352 _pt_tmp_343 = _pt_tmp_344 - _pt_tmp_351 - _pt_tmp_342 = _pt_tmp_343 / np.complex128(1 + 0j) + _pt_tmp_342 = _pt_tmp_343 / np.float64(1.0) del _pt_tmp_344, _pt_tmp_351 _pt_tmp = make_obj_array( [_pt_tmp_0, _pt_tmp_138, _pt_tmp_197, _pt_tmp_232, _pt_tmp_294, _pt_tmp_342] diff --git a/actx_dgfem_suite/suite/tiny_maxwell_3D_P3/ref_input_args.pkl.dvc b/actx_dgfem_suite/suite/tiny_maxwell_3D_P3/ref_input_args.pkl.dvc index d224638..41d6d59 100644 --- a/actx_dgfem_suite/suite/tiny_maxwell_3D_P3/ref_input_args.pkl.dvc +++ b/actx_dgfem_suite/suite/tiny_maxwell_3D_P3/ref_input_args.pkl.dvc @@ -1,5 +1,5 @@ outs: -- md5: f9cb6e204969482b0c641a6b49e34994 - size: 285859 +- md5: d588379c2d1761409f6b2dfff98b9284 + size: 156215 hash: md5 path: ref_input_args.pkl diff --git a/actx_dgfem_suite/suite/tiny_maxwell_3D_P3/ref_outputs.pkl.dvc b/actx_dgfem_suite/suite/tiny_maxwell_3D_P3/ref_outputs.pkl.dvc index 574753f..7ed4001 100644 --- a/actx_dgfem_suite/suite/tiny_maxwell_3D_P3/ref_outputs.pkl.dvc +++ b/actx_dgfem_suite/suite/tiny_maxwell_3D_P3/ref_outputs.pkl.dvc @@ -1,5 +1,5 @@ outs: -- md5: 8178849a0ef6a2ecbab923ad83786e17 - size: 311709 +- md5: fbbd15aecaff25a444d9f4d1423952ee + size: 156179 hash: md5 path: ref_outputs.pkl diff --git a/actx_dgfem_suite/suite/tiny_maxwell_3D_P4/literals.npz.dvc b/actx_dgfem_suite/suite/tiny_maxwell_3D_P4/literals.npz.dvc index 06f8130..5cd5d89 100644 --- a/actx_dgfem_suite/suite/tiny_maxwell_3D_P4/literals.npz.dvc +++ b/actx_dgfem_suite/suite/tiny_maxwell_3D_P4/literals.npz.dvc @@ -1,5 +1,5 @@ outs: -- md5: 59ac9054d095bb1ab782a7b2c7b4673c - size: 94136 +- md5: d41188ec0709febb044f1397ebd79729 + size: 77336 hash: md5 path: literals.npz diff --git a/actx_dgfem_suite/suite/tiny_maxwell_3D_P4/main.py b/actx_dgfem_suite/suite/tiny_maxwell_3D_P4/main.py index d05a8f4..efabde3 100644 --- a/actx_dgfem_suite/suite/tiny_maxwell_3D_P4/main.py +++ b/actx_dgfem_suite/suite/tiny_maxwell_3D_P4/main.py @@ -58,62 +58,15 @@ def _rhs_inner( ], ) _pt_t_unit_10 = make_loopy_program( - "{ [_0, _1] : 0 <= _0 <= 47 and 0 <= _1 <= 14 }", - "out[_0, _1] = in_0[in_1[_0, 0] % 48, in_2[_0, _1] % 35]", - [ - lp.GlobalArg("out", dtype=np.float64, shape=(48, 15)), - lp.GlobalArg("in_0", dtype=np.float64, shape=(48, 35)), - lp.GlobalArg("in_1", dtype=np.int32, shape=(48, 1)), - lp.GlobalArg("in_2", dtype=np.int64, shape=(48, 15)), - ], - ) - _pt_t_unit_11 = make_loopy_program( - "{ [_0, _1] : 0 <= _0 <= 47 and 0 <= _1 <= 14 }", - "out[_0, _1] = in_0[in_1[_0, 0] % 48, in_2[_0, _1] % 35]", - [ - lp.GlobalArg("out", dtype=np.complex128, shape=(48, 15)), - lp.GlobalArg("in_0", dtype=np.complex128, shape=(48, 35)), - lp.GlobalArg("in_1", dtype=np.int32, shape=(48, 1)), - lp.GlobalArg("in_2", dtype=np.int64, shape=(48, 15)), - ], - ) - _pt_t_unit_12 = make_loopy_program( - "{ [_0, _1] : 0 <= _0 <= 47 and 0 <= _1 <= 14 }", - "out[_0, _1] = cast(np:dtype('complex128'), _in0[_0, 0])*_in1[_0, _1]", - [ - lp.GlobalArg("out", dtype=np.complex128, shape=(48, 15)), - lp.GlobalArg("_in0", dtype=np.float64, shape=(48, 1)), - lp.GlobalArg("_in1", dtype=np.complex128, shape=(48, 15)), - ], - ) - _pt_t_unit_13 = make_loopy_program( "{ [_0, _1] : 0 <= _0 <= 191 and 0 <= _1 <= 14 }", "out[_0, _1] = in_0[in_1[_0, 0] % 48, in_2[_0, _1] % 15]", [ - lp.GlobalArg("out", dtype=np.complex128, shape=(192, 15)), - lp.GlobalArg("in_0", dtype=np.complex128, shape=(48, 15)), + lp.GlobalArg("out", dtype=np.float64, shape=(192, 15)), + lp.GlobalArg("in_0", dtype=np.float64, shape=(48, 15)), lp.GlobalArg("in_1", dtype=np.int32, shape=(192, 1)), lp.GlobalArg("in_2", dtype=np.int64, shape=(192, 15)), ], ) - _pt_t_unit_14 = make_loopy_program( - "{ [_0, _1] : 0 <= _0 <= 143 and 0 <= _1 <= 14 }", - "out[_0, _1] = _in0[_0, 0]*_in1[_0, _1]", - [ - lp.GlobalArg("out", dtype=np.float64, shape=(144, 15)), - lp.GlobalArg("_in0", dtype=np.float64, shape=(144, 1)), - lp.GlobalArg("_in1", dtype=np.float64, shape=(144, 15)), - ], - ) - _pt_t_unit_15 = make_loopy_program( - "{ [_0, _1] : 0 <= _0 <= 47 and 0 <= _1 <= 14 }", - "out[_0, _1] = _in0[_0, 0]*_in1[_0, _1]", - [ - lp.GlobalArg("out", dtype=np.float64, shape=(48, 15)), - lp.GlobalArg("_in0", dtype=np.float64, shape=(48, 1)), - lp.GlobalArg("_in1", dtype=np.float64, shape=(48, 15)), - ], - ) _pt_t_unit_2 = make_loopy_program( "{ [_0, _1] : 0 <= _0 <= 143 and 0 <= _1 <= 14 }", "out[_0, _1] = in_0[in_1[_0, 0] % 144, in_2[_0, _1] % 15]", @@ -126,34 +79,14 @@ def _rhs_inner( ) _pt_t_unit_3 = make_loopy_program( "{ [_0, _1] : 0 <= _0 <= 143 and 0 <= _1 <= 14 }", - "out[_0, _1] = in_0[in_1[_0, 0] % 48, in_2[_0, _1] % 35]", - [ - lp.GlobalArg("out", dtype=np.complex128, shape=(144, 15)), - lp.GlobalArg("in_0", dtype=np.complex128, shape=(48, 35)), - lp.GlobalArg("in_1", dtype=np.int32, shape=(144, 1)), - lp.GlobalArg("in_2", dtype=np.int64, shape=(144, 15)), - ], - ) - _pt_t_unit_4 = make_loopy_program( - "{ [_0, _1] : 0 <= _0 <= 143 and 0 <= _1 <= 14 }", - "out[_0, _1] = in_0[in_1[_0, 0] % 144, in_2[_0, _1] % 15]", - [ - lp.GlobalArg("out", dtype=np.complex128, shape=(144, 15)), - lp.GlobalArg("in_0", dtype=np.complex128, shape=(144, 15)), - lp.GlobalArg("in_1", dtype=np.int32, shape=(144, 1)), - lp.GlobalArg("in_2", dtype=np.int64, shape=(144, 15)), - ], - ) - _pt_t_unit_5 = make_loopy_program( - "{ [_0, _1] : 0 <= _0 <= 143 and 0 <= _1 <= 14 }", - "out[_0, _1] = cast(np:dtype('complex128'), _in0[_0, 0])*_in1[_0, _1]", + "out[_0, _1] = _in0[_0, 0]*_in1[_0, _1]", [ - lp.GlobalArg("out", dtype=np.complex128, shape=(144, 15)), + lp.GlobalArg("out", dtype=np.float64, shape=(144, 15)), lp.GlobalArg("_in0", dtype=np.float64, shape=(144, 1)), - lp.GlobalArg("_in1", dtype=np.complex128, shape=(144, 15)), + lp.GlobalArg("_in1", dtype=np.float64, shape=(144, 15)), ], ) - _pt_t_unit_6 = make_loopy_program( + _pt_t_unit_4 = make_loopy_program( "{ [_0, _1] : 0 <= _0 <= 191 and 0 <= _1 <= 14 }", "out[_0, _1] = in_0[in_1[_0] % 1, _1]", [ @@ -162,26 +95,26 @@ def _rhs_inner( lp.GlobalArg("in_1", dtype=np.int8, shape=(192,)), ], ) - _pt_t_unit_7 = make_loopy_program( + _pt_t_unit_5 = make_loopy_program( "{ [_0, _1] : 0 <= _0 <= 191 and 0 <= _1 <= 14 }", "out[_0, _1] = in_0[in_1[_0, 0] % 144, in_2[_0, _1] % 15]", [ - lp.GlobalArg("out", dtype=np.complex128, shape=(192, 15)), - lp.GlobalArg("in_0", dtype=np.complex128, shape=(144, 15)), + lp.GlobalArg("out", dtype=np.float64, shape=(192, 15)), + lp.GlobalArg("in_0", dtype=np.float64, shape=(144, 15)), lp.GlobalArg("in_1", dtype=np.int32, shape=(192, 1)), lp.GlobalArg("in_2", dtype=np.int64, shape=(192, 15)), ], ) - _pt_t_unit_8 = make_loopy_program( + _pt_t_unit_6 = make_loopy_program( "{ [_0, _1] : 0 <= _0 <= 191 and 0 <= _1 <= 14 }", "out[_0, _1] = _in1[_0, _1] if _in0[_0, 0] else 0", [ - lp.GlobalArg("out", dtype=np.complex128, shape=(192, 15)), + lp.GlobalArg("out", dtype=np.float64, shape=(192, 15)), lp.GlobalArg("_in0", dtype=np.int8, shape=(192, 1)), - lp.GlobalArg("_in1", dtype=np.complex128, shape=(192, 15)), + lp.GlobalArg("_in1", dtype=np.float64, shape=(192, 15)), ], ) - _pt_t_unit_9 = make_loopy_program( + _pt_t_unit_7 = make_loopy_program( "{ [_0, _1] : 0 <= _0 <= 47 and 0 <= _1 <= 14 }", "out[_0, _1] = in_0[in_1[_0] % 4, _1]", [ @@ -190,6 +123,25 @@ def _rhs_inner( lp.GlobalArg("in_1", dtype=np.int8, shape=(48,)), ], ) + _pt_t_unit_8 = make_loopy_program( + "{ [_0, _1] : 0 <= _0 <= 47 and 0 <= _1 <= 14 }", + "out[_0, _1] = in_0[in_1[_0, 0] % 48, in_2[_0, _1] % 35]", + [ + lp.GlobalArg("out", dtype=np.float64, shape=(48, 15)), + lp.GlobalArg("in_0", dtype=np.float64, shape=(48, 35)), + lp.GlobalArg("in_1", dtype=np.int32, shape=(48, 1)), + lp.GlobalArg("in_2", dtype=np.int64, shape=(48, 15)), + ], + ) + _pt_t_unit_9 = make_loopy_program( + "{ [_0, _1] : 0 <= _0 <= 47 and 0 <= _1 <= 14 }", + "out[_0, _1] = _in0[_0, 0]*_in1[_0, _1]", + [ + lp.GlobalArg("out", dtype=np.float64, shape=(48, 15)), + lp.GlobalArg("_in0", dtype=np.float64, shape=(48, 1)), + lp.GlobalArg("_in1", dtype=np.float64, shape=(48, 15)), + ], + ) _pt_data = actx.thaw(npzfile["_pt_data"]) _pt_tmp_9 = _pt_data[1] _pt_tmp_8 = _pt_tmp_9[:, :, 0] @@ -206,10 +158,10 @@ def _rhs_inner( _pt_tmp_11 = actx.einsum("ij,ikl,jl->jk", _pt_tmp_12, _pt_data_0, _actx_in_1_4_0) del _pt_tmp_13 _pt_tmp_11 = actx.tag((FirstAxisIsElementsTag(),), _pt_tmp_11) - _pt_tmp_10 = (-1 + 0j) * _pt_tmp_11 + _pt_tmp_10 = -1.0 * _pt_tmp_11 _pt_tmp_4 = _pt_tmp_5 + _pt_tmp_10 del _pt_tmp_11 - _pt_tmp_3 = 0j - _pt_tmp_4 + _pt_tmp_3 = 0.0 - _pt_tmp_4 del _pt_tmp_10, _pt_tmp_5 _pt_tmp_2 = -1 * _pt_tmp_3 del _pt_tmp_4 @@ -278,75 +230,75 @@ def _rhs_inner( _actx_in_1_1_0[_pt_tmp_39, _pt_tmp_40] if actx.permits_advanced_indexing else actx.call_loopy( - _pt_t_unit_3, in_0=_actx_in_1_1_0, in_1=_pt_tmp_39, in_2=_pt_tmp_40 + _pt_t_unit_0, in_0=_actx_in_1_1_0, in_1=_pt_tmp_39, in_2=_pt_tmp_40 )["out"] ) _pt_tmp_52 = actx.tag_axis(0, (DiscretizationElementAxisTag(),), _pt_tmp_52) _pt_tmp_52 = actx.tag_axis(1, (DiscretizationDOFAxisTag(),), _pt_tmp_52) - _pt_tmp_51 = 0j + _pt_tmp_52 + _pt_tmp_51 = 0.0 + _pt_tmp_52 _pt_tmp_50 = ( _pt_tmp_51[_pt_tmp_41, _pt_tmp_42] if actx.permits_advanced_indexing else actx.call_loopy( - _pt_t_unit_4, in_0=_pt_tmp_51, in_1=_pt_tmp_41, in_2=_pt_tmp_42 + _pt_t_unit_2, in_0=_pt_tmp_51, in_1=_pt_tmp_41, in_2=_pt_tmp_42 )["out"] ) del _pt_tmp_52 _pt_tmp_50 = actx.tag_axis(0, (DiscretizationElementAxisTag(),), _pt_tmp_50) _pt_tmp_50 = actx.tag_axis(1, (DiscretizationDOFAxisTag(),), _pt_tmp_50) - _pt_tmp_49 = 0j + _pt_tmp_50 + _pt_tmp_49 = 0.0 + _pt_tmp_50 _pt_tmp_48 = _pt_tmp_49 - _pt_tmp_51 del _pt_tmp_50 _pt_tmp_46 = ( _pt_tmp_47 * _pt_tmp_48 if actx.supports_nonscalar_broadcasting - else actx.call_loopy(_pt_t_unit_5, _in0=_pt_tmp_47, _in1=_pt_tmp_48)["out"] + else actx.call_loopy(_pt_t_unit_3, _in0=_pt_tmp_47, _in1=_pt_tmp_48)["out"] ) del _pt_tmp_49, _pt_tmp_51 - _pt_tmp_45 = 0j + _pt_tmp_46 + _pt_tmp_45 = 0.0 + _pt_tmp_46 _pt_tmp_54 = -1.0 * _pt_data_6 del _pt_tmp_46 _pt_tmp_59 = ( _actx_in_1_0_0[_pt_tmp_39, _pt_tmp_40] if actx.permits_advanced_indexing else actx.call_loopy( - _pt_t_unit_3, in_0=_actx_in_1_0_0, in_1=_pt_tmp_39, in_2=_pt_tmp_40 + _pt_t_unit_0, in_0=_actx_in_1_0_0, in_1=_pt_tmp_39, in_2=_pt_tmp_40 )["out"] ) _pt_tmp_59 = actx.tag_axis(0, (DiscretizationElementAxisTag(),), _pt_tmp_59) _pt_tmp_59 = actx.tag_axis(1, (DiscretizationDOFAxisTag(),), _pt_tmp_59) - _pt_tmp_58 = 0j + _pt_tmp_59 + _pt_tmp_58 = 0.0 + _pt_tmp_59 _pt_tmp_57 = ( _pt_tmp_58[_pt_tmp_41, _pt_tmp_42] if actx.permits_advanced_indexing else actx.call_loopy( - _pt_t_unit_4, in_0=_pt_tmp_58, in_1=_pt_tmp_41, in_2=_pt_tmp_42 + _pt_t_unit_2, in_0=_pt_tmp_58, in_1=_pt_tmp_41, in_2=_pt_tmp_42 )["out"] ) del _pt_tmp_59 _pt_tmp_57 = actx.tag_axis(0, (DiscretizationElementAxisTag(),), _pt_tmp_57) _pt_tmp_57 = actx.tag_axis(1, (DiscretizationDOFAxisTag(),), _pt_tmp_57) - _pt_tmp_56 = 0j + _pt_tmp_57 + _pt_tmp_56 = 0.0 + _pt_tmp_57 _pt_tmp_55 = _pt_tmp_56 - _pt_tmp_58 del _pt_tmp_57 _pt_tmp_53 = ( _pt_tmp_54 * _pt_tmp_55 if actx.supports_nonscalar_broadcasting - else actx.call_loopy(_pt_t_unit_5, _in0=_pt_tmp_54, _in1=_pt_tmp_55)["out"] + else actx.call_loopy(_pt_t_unit_3, _in0=_pt_tmp_54, _in1=_pt_tmp_55)["out"] ) del _pt_tmp_56, _pt_tmp_58 _pt_tmp_44 = _pt_tmp_45 + _pt_tmp_53 - _pt_tmp_43 = (0.5 + 0j) * _pt_tmp_44 + _pt_tmp_43 = 0.5 * _pt_tmp_44 del _pt_tmp_45, _pt_tmp_53 _pt_tmp_32 = _pt_tmp_33 - _pt_tmp_43 del _pt_tmp_44 _pt_tmp_30 = ( _pt_tmp_31 * _pt_tmp_32 if actx.supports_nonscalar_broadcasting - else actx.call_loopy(_pt_t_unit_5, _in0=_pt_tmp_31, _in1=_pt_tmp_32)["out"] + else actx.call_loopy(_pt_t_unit_3, _in0=_pt_tmp_31, _in1=_pt_tmp_32)["out"] ) del _pt_tmp_33, _pt_tmp_43 - _pt_tmp_29 = 0j + _pt_tmp_30 + _pt_tmp_29 = 0.0 + _pt_tmp_30 _pt_data_14 = actx.thaw(npzfile["_pt_data_14"]) del _pt_tmp_30 _pt_tmp_61 = -1.0 * _pt_data_14 @@ -354,78 +306,78 @@ def _rhs_inner( _actx_in_1_4_0[_pt_tmp_39, _pt_tmp_40] if actx.permits_advanced_indexing else actx.call_loopy( - _pt_t_unit_3, in_0=_actx_in_1_4_0, in_1=_pt_tmp_39, in_2=_pt_tmp_40 + _pt_t_unit_0, in_0=_actx_in_1_4_0, in_1=_pt_tmp_39, in_2=_pt_tmp_40 )["out"] ) _pt_tmp_68 = actx.tag_axis(0, (DiscretizationElementAxisTag(),), _pt_tmp_68) _pt_tmp_68 = actx.tag_axis(1, (DiscretizationDOFAxisTag(),), _pt_tmp_68) - _pt_tmp_67 = 0j + _pt_tmp_68 + _pt_tmp_67 = 0.0 + _pt_tmp_68 _pt_tmp_66 = ( _pt_tmp_67[_pt_tmp_41, _pt_tmp_42] if actx.permits_advanced_indexing else actx.call_loopy( - _pt_t_unit_4, in_0=_pt_tmp_67, in_1=_pt_tmp_41, in_2=_pt_tmp_42 + _pt_t_unit_2, in_0=_pt_tmp_67, in_1=_pt_tmp_41, in_2=_pt_tmp_42 )["out"] ) del _pt_tmp_68 _pt_tmp_66 = actx.tag_axis(0, (DiscretizationElementAxisTag(),), _pt_tmp_66) _pt_tmp_66 = actx.tag_axis(1, (DiscretizationDOFAxisTag(),), _pt_tmp_66) - _pt_tmp_65 = 0j + _pt_tmp_66 + _pt_tmp_65 = 0.0 + _pt_tmp_66 _pt_tmp_64 = _pt_tmp_65 - _pt_tmp_67 del _pt_tmp_66 - _pt_tmp_63 = (1 + 0j) * _pt_tmp_64 + _pt_tmp_63 = 1.0 * _pt_tmp_64 del _pt_tmp_65, _pt_tmp_67 _pt_tmp_73 = -1.0 * _pt_data_13 _pt_tmp_78 = ( _actx_in_1_2_0[_pt_tmp_39, _pt_tmp_40] if actx.permits_advanced_indexing else actx.call_loopy( - _pt_t_unit_3, in_0=_actx_in_1_2_0, in_1=_pt_tmp_39, in_2=_pt_tmp_40 + _pt_t_unit_0, in_0=_actx_in_1_2_0, in_1=_pt_tmp_39, in_2=_pt_tmp_40 )["out"] ) _pt_tmp_78 = actx.tag_axis(0, (DiscretizationElementAxisTag(),), _pt_tmp_78) _pt_tmp_78 = actx.tag_axis(1, (DiscretizationDOFAxisTag(),), _pt_tmp_78) - _pt_tmp_77 = 0j + _pt_tmp_78 + _pt_tmp_77 = 0.0 + _pt_tmp_78 _pt_tmp_76 = ( _pt_tmp_77[_pt_tmp_41, _pt_tmp_42] if actx.permits_advanced_indexing else actx.call_loopy( - _pt_t_unit_4, in_0=_pt_tmp_77, in_1=_pt_tmp_41, in_2=_pt_tmp_42 + _pt_t_unit_2, in_0=_pt_tmp_77, in_1=_pt_tmp_41, in_2=_pt_tmp_42 )["out"] ) del _pt_tmp_78 _pt_tmp_76 = actx.tag_axis(0, (DiscretizationElementAxisTag(),), _pt_tmp_76) _pt_tmp_76 = actx.tag_axis(1, (DiscretizationDOFAxisTag(),), _pt_tmp_76) - _pt_tmp_75 = 0j + _pt_tmp_76 + _pt_tmp_75 = 0.0 + _pt_tmp_76 _pt_tmp_74 = _pt_tmp_75 - _pt_tmp_77 del _pt_tmp_76 _pt_tmp_72 = ( _pt_tmp_73 * _pt_tmp_74 if actx.supports_nonscalar_broadcasting - else actx.call_loopy(_pt_t_unit_5, _in0=_pt_tmp_73, _in1=_pt_tmp_74)["out"] + else actx.call_loopy(_pt_t_unit_3, _in0=_pt_tmp_73, _in1=_pt_tmp_74)["out"] ) del _pt_tmp_75, _pt_tmp_77 - _pt_tmp_71 = 0j + _pt_tmp_72 + _pt_tmp_71 = 0.0 + _pt_tmp_72 _pt_tmp_80 = 1.0 * _pt_data_14 del _pt_tmp_72 _pt_tmp_79 = ( _pt_tmp_80 * _pt_tmp_55 if actx.supports_nonscalar_broadcasting - else actx.call_loopy(_pt_t_unit_5, _in0=_pt_tmp_80, _in1=_pt_tmp_55)["out"] + else actx.call_loopy(_pt_t_unit_3, _in0=_pt_tmp_80, _in1=_pt_tmp_55)["out"] ) _pt_tmp_70 = _pt_tmp_71 + _pt_tmp_79 - _pt_tmp_69 = (0.5 + 0j) * _pt_tmp_70 + _pt_tmp_69 = 0.5 * _pt_tmp_70 del _pt_tmp_71, _pt_tmp_79 _pt_tmp_62 = _pt_tmp_63 - _pt_tmp_69 del _pt_tmp_70 _pt_tmp_60 = ( _pt_tmp_61 * _pt_tmp_62 if actx.supports_nonscalar_broadcasting - else actx.call_loopy(_pt_t_unit_5, _in0=_pt_tmp_61, _in1=_pt_tmp_62)["out"] + else actx.call_loopy(_pt_t_unit_3, _in0=_pt_tmp_61, _in1=_pt_tmp_62)["out"] ) del _pt_tmp_63, _pt_tmp_69 _pt_tmp_28 = _pt_tmp_29 + _pt_tmp_60 - _pt_tmp_27 = (-0.5 + 0j) * _pt_tmp_28 + _pt_tmp_27 = -0.5 * _pt_tmp_28 del _pt_tmp_29, _pt_tmp_60 _pt_data_15 = actx.thaw(npzfile["_pt_data_15"]) del _pt_tmp_28 @@ -435,26 +387,26 @@ def _rhs_inner( _pt_tmp_82 = ( _pt_data_16[_pt_data_17] if actx.permits_advanced_indexing - else actx.call_loopy(_pt_t_unit_6, in_0=_pt_data_16, in_1=_pt_data_17)["out"] + else actx.call_loopy(_pt_t_unit_4, in_0=_pt_data_16, in_1=_pt_data_17)["out"] ) _pt_tmp_26 = ( _pt_tmp_27[_pt_tmp_81, _pt_tmp_82] if actx.permits_advanced_indexing else actx.call_loopy( - _pt_t_unit_7, in_0=_pt_tmp_27, in_1=_pt_tmp_81, in_2=_pt_tmp_82 + _pt_t_unit_5, in_0=_pt_tmp_27, in_1=_pt_tmp_81, in_2=_pt_tmp_82 )["out"] ) _pt_tmp_24 = ( actx.np.where(_pt_tmp_25, _pt_tmp_26, 0) if actx.supports_nonscalar_broadcasting - else actx.call_loopy(_pt_t_unit_8, _in0=_pt_tmp_25, _in1=_pt_tmp_26)["out"] + else actx.call_loopy(_pt_t_unit_6, _in0=_pt_tmp_25, _in1=_pt_tmp_26)["out"] ) del _pt_tmp_27 _pt_tmp_24 = actx.tag_axis(0, (DiscretizationElementAxisTag(),), _pt_tmp_24) del _pt_tmp_26 _pt_tmp_24 = actx.tag_axis(1, (DiscretizationDOFAxisTag(),), _pt_tmp_24) - _pt_tmp_23 = 0j + _pt_tmp_24 - _pt_tmp_22 = 0j + _pt_tmp_23 + _pt_tmp_23 = 0.0 + _pt_tmp_24 + _pt_tmp_22 = 0.0 + _pt_tmp_23 del _pt_tmp_24 _pt_data_18 = actx.thaw(npzfile["_pt_data_18"]) del _pt_tmp_23 @@ -467,13 +419,13 @@ def _rhs_inner( _pt_tmp_102 = ( _pt_data_8[_pt_data_21] if actx.permits_advanced_indexing - else actx.call_loopy(_pt_t_unit_9, in_0=_pt_data_8, in_1=_pt_data_21)["out"] + else actx.call_loopy(_pt_t_unit_7, in_0=_pt_data_8, in_1=_pt_data_21)["out"] ) _pt_tmp_100 = ( _actx_in_1_5_0[_pt_tmp_101, _pt_tmp_102] if actx.permits_advanced_indexing else actx.call_loopy( - _pt_t_unit_10, in_0=_actx_in_1_5_0, in_1=_pt_tmp_101, in_2=_pt_tmp_102 + _pt_t_unit_8, in_0=_actx_in_1_5_0, in_1=_pt_tmp_101, in_2=_pt_tmp_102 )["out"] ) _pt_tmp_100 = actx.tag_axis(0, (DiscretizationElementAxisTag(),), _pt_tmp_100) @@ -489,59 +441,55 @@ def _rhs_inner( _actx_in_1_1_0[_pt_tmp_101, _pt_tmp_102] if actx.permits_advanced_indexing else actx.call_loopy( - _pt_t_unit_11, in_0=_actx_in_1_1_0, in_1=_pt_tmp_101, in_2=_pt_tmp_102 + _pt_t_unit_8, in_0=_actx_in_1_1_0, in_1=_pt_tmp_101, in_2=_pt_tmp_102 )["out"] ) _pt_tmp_111 = actx.tag_axis(0, (DiscretizationElementAxisTag(),), _pt_tmp_111) _pt_tmp_111 = actx.tag_axis(1, (DiscretizationDOFAxisTag(),), _pt_tmp_111) - _pt_tmp_110 = 0j + _pt_tmp_111 + _pt_tmp_110 = 0.0 + _pt_tmp_111 _pt_tmp_109 = -1 * _pt_tmp_110 del _pt_tmp_111 _pt_tmp_108 = _pt_tmp_109 - _pt_tmp_110 _pt_tmp_106 = ( _pt_tmp_107 * _pt_tmp_108 if actx.supports_nonscalar_broadcasting - else actx.call_loopy(_pt_t_unit_12, _in0=_pt_tmp_107, _in1=_pt_tmp_108)[ - "out" - ] + else actx.call_loopy(_pt_t_unit_9, _in0=_pt_tmp_107, _in1=_pt_tmp_108)["out"] ) del _pt_tmp_109, _pt_tmp_110 - _pt_tmp_105 = 0j + _pt_tmp_106 + _pt_tmp_105 = 0.0 + _pt_tmp_106 _pt_tmp_113 = -1.0 * _pt_data_19 del _pt_tmp_106 _pt_tmp_117 = ( _actx_in_1_0_0[_pt_tmp_101, _pt_tmp_102] if actx.permits_advanced_indexing else actx.call_loopy( - _pt_t_unit_11, in_0=_actx_in_1_0_0, in_1=_pt_tmp_101, in_2=_pt_tmp_102 + _pt_t_unit_8, in_0=_actx_in_1_0_0, in_1=_pt_tmp_101, in_2=_pt_tmp_102 )["out"] ) _pt_tmp_117 = actx.tag_axis(0, (DiscretizationElementAxisTag(),), _pt_tmp_117) _pt_tmp_117 = actx.tag_axis(1, (DiscretizationDOFAxisTag(),), _pt_tmp_117) - _pt_tmp_116 = 0j + _pt_tmp_117 + _pt_tmp_116 = 0.0 + _pt_tmp_117 _pt_tmp_115 = -1 * _pt_tmp_116 del _pt_tmp_117 _pt_tmp_114 = _pt_tmp_115 - _pt_tmp_116 _pt_tmp_112 = ( _pt_tmp_113 * _pt_tmp_114 if actx.supports_nonscalar_broadcasting - else actx.call_loopy(_pt_t_unit_12, _in0=_pt_tmp_113, _in1=_pt_tmp_114)[ - "out" - ] + else actx.call_loopy(_pt_t_unit_9, _in0=_pt_tmp_113, _in1=_pt_tmp_114)["out"] ) del _pt_tmp_115, _pt_tmp_116 _pt_tmp_104 = _pt_tmp_105 + _pt_tmp_112 - _pt_tmp_103 = (0.5 + 0j) * _pt_tmp_104 + _pt_tmp_103 = 0.5 * _pt_tmp_104 del _pt_tmp_105, _pt_tmp_112 _pt_tmp_96 = _pt_tmp_97 - _pt_tmp_103 del _pt_tmp_104 _pt_tmp_94 = ( _pt_tmp_95 * _pt_tmp_96 if actx.supports_nonscalar_broadcasting - else actx.call_loopy(_pt_t_unit_12, _in0=_pt_tmp_95, _in1=_pt_tmp_96)["out"] + else actx.call_loopy(_pt_t_unit_9, _in0=_pt_tmp_95, _in1=_pt_tmp_96)["out"] ) del _pt_tmp_103, _pt_tmp_97 - _pt_tmp_93 = 0j + _pt_tmp_94 + _pt_tmp_93 = 0.0 + _pt_tmp_94 _pt_data_23 = actx.thaw(npzfile["_pt_data_23"]) del _pt_tmp_94 _pt_tmp_119 = -1.0 * _pt_data_23 @@ -549,63 +497,57 @@ def _rhs_inner( _actx_in_1_4_0[_pt_tmp_101, _pt_tmp_102] if actx.permits_advanced_indexing else actx.call_loopy( - _pt_t_unit_11, in_0=_actx_in_1_4_0, in_1=_pt_tmp_101, in_2=_pt_tmp_102 + _pt_t_unit_8, in_0=_actx_in_1_4_0, in_1=_pt_tmp_101, in_2=_pt_tmp_102 )["out"] ) _pt_tmp_124 = actx.tag_axis(0, (DiscretizationElementAxisTag(),), _pt_tmp_124) _pt_tmp_124 = actx.tag_axis(1, (DiscretizationDOFAxisTag(),), _pt_tmp_124) - _pt_tmp_123 = 0j + _pt_tmp_124 + _pt_tmp_123 = 0.0 + _pt_tmp_124 _pt_tmp_122 = _pt_tmp_123 - _pt_tmp_123 del _pt_tmp_124 - _pt_tmp_121 = (1 + 0j) * _pt_tmp_122 + _pt_tmp_121 = 1.0 * _pt_tmp_122 del _pt_tmp_123 _pt_tmp_129 = -1.0 * _pt_data_22 _pt_tmp_133 = ( _actx_in_1_2_0[_pt_tmp_101, _pt_tmp_102] if actx.permits_advanced_indexing else actx.call_loopy( - _pt_t_unit_11, in_0=_actx_in_1_2_0, in_1=_pt_tmp_101, in_2=_pt_tmp_102 + _pt_t_unit_8, in_0=_actx_in_1_2_0, in_1=_pt_tmp_101, in_2=_pt_tmp_102 )["out"] ) _pt_tmp_133 = actx.tag_axis(0, (DiscretizationElementAxisTag(),), _pt_tmp_133) _pt_tmp_133 = actx.tag_axis(1, (DiscretizationDOFAxisTag(),), _pt_tmp_133) - _pt_tmp_132 = 0j + _pt_tmp_133 + _pt_tmp_132 = 0.0 + _pt_tmp_133 _pt_tmp_131 = -1 * _pt_tmp_132 del _pt_tmp_133 _pt_tmp_130 = _pt_tmp_131 - _pt_tmp_132 _pt_tmp_128 = ( _pt_tmp_129 * _pt_tmp_130 if actx.supports_nonscalar_broadcasting - else actx.call_loopy(_pt_t_unit_12, _in0=_pt_tmp_129, _in1=_pt_tmp_130)[ - "out" - ] + else actx.call_loopy(_pt_t_unit_9, _in0=_pt_tmp_129, _in1=_pt_tmp_130)["out"] ) del _pt_tmp_131, _pt_tmp_132 - _pt_tmp_127 = 0j + _pt_tmp_128 + _pt_tmp_127 = 0.0 + _pt_tmp_128 _pt_tmp_135 = 1.0 * _pt_data_23 del _pt_tmp_128 _pt_tmp_134 = ( _pt_tmp_135 * _pt_tmp_114 if actx.supports_nonscalar_broadcasting - else actx.call_loopy(_pt_t_unit_12, _in0=_pt_tmp_135, _in1=_pt_tmp_114)[ - "out" - ] + else actx.call_loopy(_pt_t_unit_9, _in0=_pt_tmp_135, _in1=_pt_tmp_114)["out"] ) _pt_tmp_126 = _pt_tmp_127 + _pt_tmp_134 - _pt_tmp_125 = (0.5 + 0j) * _pt_tmp_126 + _pt_tmp_125 = 0.5 * _pt_tmp_126 del _pt_tmp_127, _pt_tmp_134 _pt_tmp_120 = _pt_tmp_121 - _pt_tmp_125 del _pt_tmp_126 _pt_tmp_118 = ( _pt_tmp_119 * _pt_tmp_120 if actx.supports_nonscalar_broadcasting - else actx.call_loopy(_pt_t_unit_12, _in0=_pt_tmp_119, _in1=_pt_tmp_120)[ - "out" - ] + else actx.call_loopy(_pt_t_unit_9, _in0=_pt_tmp_119, _in1=_pt_tmp_120)["out"] ) del _pt_tmp_121, _pt_tmp_125 _pt_tmp_92 = _pt_tmp_93 + _pt_tmp_118 - _pt_tmp_91 = (-0.5 + 0j) * _pt_tmp_92 + _pt_tmp_91 = -0.5 * _pt_tmp_92 del _pt_tmp_118, _pt_tmp_93 _pt_data_24 = actx.thaw(npzfile["_pt_data_24"]) del _pt_tmp_92 @@ -614,22 +556,22 @@ def _rhs_inner( _pt_tmp_91[_pt_tmp_136, _pt_tmp_82] if actx.permits_advanced_indexing else actx.call_loopy( - _pt_t_unit_13, in_0=_pt_tmp_91, in_1=_pt_tmp_136, in_2=_pt_tmp_82 + _pt_t_unit_10, in_0=_pt_tmp_91, in_1=_pt_tmp_136, in_2=_pt_tmp_82 )["out"] ) _pt_tmp_88 = ( actx.np.where(_pt_tmp_89, _pt_tmp_90, 0) if actx.supports_nonscalar_broadcasting - else actx.call_loopy(_pt_t_unit_8, _in0=_pt_tmp_89, _in1=_pt_tmp_90)["out"] + else actx.call_loopy(_pt_t_unit_6, _in0=_pt_tmp_89, _in1=_pt_tmp_90)["out"] ) del _pt_tmp_91 _pt_tmp_88 = actx.tag_axis(0, (DiscretizationElementAxisTag(),), _pt_tmp_88) del _pt_tmp_90 _pt_tmp_88 = actx.tag_axis(1, (DiscretizationDOFAxisTag(),), _pt_tmp_88) - _pt_tmp_87 = 0j + _pt_tmp_88 - _pt_tmp_86 = 0j + _pt_tmp_87 + _pt_tmp_87 = 0.0 + _pt_tmp_88 + _pt_tmp_86 = 0.0 + _pt_tmp_87 del _pt_tmp_88 - _pt_tmp_137 = actx.np.zeros((192, 15), dtype=np.complex128) + _pt_tmp_137 = actx.np.zeros((192, 15), dtype=np.float64) del _pt_tmp_87 _pt_tmp_85 = _pt_tmp_86 + _pt_tmp_137 _pt_tmp_84 = _pt_tmp_85 + _pt_tmp_137 @@ -649,7 +591,7 @@ def _rhs_inner( _pt_tmp_14 = actx.tag((FirstAxisIsElementsTag(),), _pt_tmp_14) del _pt_tmp_17 _pt_tmp_1 = _pt_tmp_2 - _pt_tmp_14 - _pt_tmp_0 = _pt_tmp_1 / np.complex128(1 + 0j) + _pt_tmp_0 = _pt_tmp_1 / np.float64(1.0) del _pt_tmp_14, _pt_tmp_2 _pt_tmp_147 = _pt_data[0] del _pt_tmp_1 @@ -667,168 +609,164 @@ def _rhs_inner( ) del _pt_tmp_144 _pt_tmp_149 = actx.tag((FirstAxisIsElementsTag(),), _pt_tmp_149) - _pt_tmp_148 = (1 + 0j) * _pt_tmp_149 + _pt_tmp_148 = 1.0 * _pt_tmp_149 _pt_tmp_142 = _pt_tmp_143 + _pt_tmp_148 del _pt_tmp_149 - _pt_tmp_141 = 0j - _pt_tmp_142 + _pt_tmp_141 = 0.0 - _pt_tmp_142 del _pt_tmp_143, _pt_tmp_148 _pt_tmp_140 = -1 * _pt_tmp_141 del _pt_tmp_142 _pt_tmp_161 = ( _pt_tmp_73 * _pt_tmp_32 if actx.supports_nonscalar_broadcasting - else actx.call_loopy(_pt_t_unit_5, _in0=_pt_tmp_73, _in1=_pt_tmp_32)["out"] + else actx.call_loopy(_pt_t_unit_3, _in0=_pt_tmp_73, _in1=_pt_tmp_32)["out"] ) del _pt_tmp_141 - _pt_tmp_160 = 0j + _pt_tmp_161 + _pt_tmp_160 = 0.0 + _pt_tmp_161 del _pt_tmp_32 _pt_tmp_169 = ( _actx_in_1_3_0[_pt_tmp_39, _pt_tmp_40] if actx.permits_advanced_indexing else actx.call_loopy( - _pt_t_unit_3, in_0=_actx_in_1_3_0, in_1=_pt_tmp_39, in_2=_pt_tmp_40 + _pt_t_unit_0, in_0=_actx_in_1_3_0, in_1=_pt_tmp_39, in_2=_pt_tmp_40 )["out"] ) del _pt_tmp_161 _pt_tmp_169 = actx.tag_axis(0, (DiscretizationElementAxisTag(),), _pt_tmp_169) del _pt_tmp_39, _pt_tmp_40 _pt_tmp_169 = actx.tag_axis(1, (DiscretizationDOFAxisTag(),), _pt_tmp_169) - _pt_tmp_168 = 0j + _pt_tmp_169 + _pt_tmp_168 = 0.0 + _pt_tmp_169 _pt_tmp_167 = ( _pt_tmp_168[_pt_tmp_41, _pt_tmp_42] if actx.permits_advanced_indexing else actx.call_loopy( - _pt_t_unit_4, in_0=_pt_tmp_168, in_1=_pt_tmp_41, in_2=_pt_tmp_42 + _pt_t_unit_2, in_0=_pt_tmp_168, in_1=_pt_tmp_41, in_2=_pt_tmp_42 )["out"] ) del _pt_tmp_169 _pt_tmp_167 = actx.tag_axis(0, (DiscretizationElementAxisTag(),), _pt_tmp_167) del _pt_tmp_41, _pt_tmp_42 _pt_tmp_167 = actx.tag_axis(1, (DiscretizationDOFAxisTag(),), _pt_tmp_167) - _pt_tmp_166 = 0j + _pt_tmp_167 + _pt_tmp_166 = 0.0 + _pt_tmp_167 _pt_tmp_165 = _pt_tmp_166 - _pt_tmp_168 del _pt_tmp_167 - _pt_tmp_164 = (1 + 0j) * _pt_tmp_165 + _pt_tmp_164 = 1.0 * _pt_tmp_165 del _pt_tmp_166, _pt_tmp_168 _pt_tmp_173 = ( _pt_tmp_31 * _pt_tmp_74 if actx.supports_nonscalar_broadcasting - else actx.call_loopy(_pt_t_unit_5, _in0=_pt_tmp_31, _in1=_pt_tmp_74)["out"] + else actx.call_loopy(_pt_t_unit_3, _in0=_pt_tmp_31, _in1=_pt_tmp_74)["out"] ) - _pt_tmp_172 = 0j + _pt_tmp_173 + _pt_tmp_172 = 0.0 + _pt_tmp_173 _pt_tmp_174 = ( _pt_tmp_61 * _pt_tmp_48 if actx.supports_nonscalar_broadcasting - else actx.call_loopy(_pt_t_unit_5, _in0=_pt_tmp_61, _in1=_pt_tmp_48)["out"] + else actx.call_loopy(_pt_t_unit_3, _in0=_pt_tmp_61, _in1=_pt_tmp_48)["out"] ) del _pt_tmp_173 _pt_tmp_171 = _pt_tmp_172 + _pt_tmp_174 - _pt_tmp_170 = (0.5 + 0j) * _pt_tmp_171 + _pt_tmp_170 = 0.5 * _pt_tmp_171 del _pt_tmp_172, _pt_tmp_174 _pt_tmp_163 = _pt_tmp_164 - _pt_tmp_170 del _pt_tmp_171 _pt_tmp_162 = ( _pt_tmp_80 * _pt_tmp_163 if actx.supports_nonscalar_broadcasting - else actx.call_loopy(_pt_t_unit_5, _in0=_pt_tmp_80, _in1=_pt_tmp_163)["out"] + else actx.call_loopy(_pt_t_unit_3, _in0=_pt_tmp_80, _in1=_pt_tmp_163)["out"] ) del _pt_tmp_164, _pt_tmp_170 _pt_tmp_159 = _pt_tmp_160 + _pt_tmp_162 - _pt_tmp_158 = (-0.5 + 0j) * _pt_tmp_159 + _pt_tmp_158 = -0.5 * _pt_tmp_159 del _pt_tmp_160, _pt_tmp_162 _pt_tmp_157 = ( _pt_tmp_158[_pt_tmp_81, _pt_tmp_82] if actx.permits_advanced_indexing else actx.call_loopy( - _pt_t_unit_7, in_0=_pt_tmp_158, in_1=_pt_tmp_81, in_2=_pt_tmp_82 + _pt_t_unit_5, in_0=_pt_tmp_158, in_1=_pt_tmp_81, in_2=_pt_tmp_82 )["out"] ) del _pt_tmp_159 _pt_tmp_156 = ( actx.np.where(_pt_tmp_25, _pt_tmp_157, 0) if actx.supports_nonscalar_broadcasting - else actx.call_loopy(_pt_t_unit_8, _in0=_pt_tmp_25, _in1=_pt_tmp_157)["out"] + else actx.call_loopy(_pt_t_unit_6, _in0=_pt_tmp_25, _in1=_pt_tmp_157)["out"] ) del _pt_tmp_158 _pt_tmp_156 = actx.tag_axis(0, (DiscretizationElementAxisTag(),), _pt_tmp_156) del _pt_tmp_157 _pt_tmp_156 = actx.tag_axis(1, (DiscretizationDOFAxisTag(),), _pt_tmp_156) - _pt_tmp_155 = 0j + _pt_tmp_156 - _pt_tmp_154 = 0j + _pt_tmp_155 + _pt_tmp_155 = 0.0 + _pt_tmp_156 + _pt_tmp_154 = 0.0 + _pt_tmp_155 del _pt_tmp_156 _pt_tmp_185 = ( _pt_tmp_129 * _pt_tmp_96 if actx.supports_nonscalar_broadcasting - else actx.call_loopy(_pt_t_unit_12, _in0=_pt_tmp_129, _in1=_pt_tmp_96)["out"] + else actx.call_loopy(_pt_t_unit_9, _in0=_pt_tmp_129, _in1=_pt_tmp_96)["out"] ) del _pt_tmp_155 - _pt_tmp_184 = 0j + _pt_tmp_185 + _pt_tmp_184 = 0.0 + _pt_tmp_185 del _pt_tmp_96 _pt_tmp_191 = ( _actx_in_1_3_0[_pt_tmp_101, _pt_tmp_102] if actx.permits_advanced_indexing else actx.call_loopy( - _pt_t_unit_11, in_0=_actx_in_1_3_0, in_1=_pt_tmp_101, in_2=_pt_tmp_102 + _pt_t_unit_8, in_0=_actx_in_1_3_0, in_1=_pt_tmp_101, in_2=_pt_tmp_102 )["out"] ) del _pt_tmp_185 _pt_tmp_191 = actx.tag_axis(0, (DiscretizationElementAxisTag(),), _pt_tmp_191) del _pt_tmp_101, _pt_tmp_102 _pt_tmp_191 = actx.tag_axis(1, (DiscretizationDOFAxisTag(),), _pt_tmp_191) - _pt_tmp_190 = 0j + _pt_tmp_191 + _pt_tmp_190 = 0.0 + _pt_tmp_191 _pt_tmp_189 = _pt_tmp_190 - _pt_tmp_190 del _pt_tmp_191 - _pt_tmp_188 = (1 + 0j) * _pt_tmp_189 + _pt_tmp_188 = 1.0 * _pt_tmp_189 del _pt_tmp_190 _pt_tmp_195 = ( _pt_tmp_95 * _pt_tmp_130 if actx.supports_nonscalar_broadcasting - else actx.call_loopy(_pt_t_unit_12, _in0=_pt_tmp_95, _in1=_pt_tmp_130)["out"] + else actx.call_loopy(_pt_t_unit_9, _in0=_pt_tmp_95, _in1=_pt_tmp_130)["out"] ) - _pt_tmp_194 = 0j + _pt_tmp_195 + _pt_tmp_194 = 0.0 + _pt_tmp_195 _pt_tmp_196 = ( _pt_tmp_119 * _pt_tmp_108 if actx.supports_nonscalar_broadcasting - else actx.call_loopy(_pt_t_unit_12, _in0=_pt_tmp_119, _in1=_pt_tmp_108)[ - "out" - ] + else actx.call_loopy(_pt_t_unit_9, _in0=_pt_tmp_119, _in1=_pt_tmp_108)["out"] ) del _pt_tmp_195 _pt_tmp_193 = _pt_tmp_194 + _pt_tmp_196 - _pt_tmp_192 = (0.5 + 0j) * _pt_tmp_193 + _pt_tmp_192 = 0.5 * _pt_tmp_193 del _pt_tmp_194, _pt_tmp_196 _pt_tmp_187 = _pt_tmp_188 - _pt_tmp_192 del _pt_tmp_193 _pt_tmp_186 = ( _pt_tmp_135 * _pt_tmp_187 if actx.supports_nonscalar_broadcasting - else actx.call_loopy(_pt_t_unit_12, _in0=_pt_tmp_135, _in1=_pt_tmp_187)[ - "out" - ] + else actx.call_loopy(_pt_t_unit_9, _in0=_pt_tmp_135, _in1=_pt_tmp_187)["out"] ) del _pt_tmp_188, _pt_tmp_192 _pt_tmp_183 = _pt_tmp_184 + _pt_tmp_186 - _pt_tmp_182 = (-0.5 + 0j) * _pt_tmp_183 + _pt_tmp_182 = -0.5 * _pt_tmp_183 del _pt_tmp_184, _pt_tmp_186 _pt_tmp_181 = ( _pt_tmp_182[_pt_tmp_136, _pt_tmp_82] if actx.permits_advanced_indexing else actx.call_loopy( - _pt_t_unit_13, in_0=_pt_tmp_182, in_1=_pt_tmp_136, in_2=_pt_tmp_82 + _pt_t_unit_10, in_0=_pt_tmp_182, in_1=_pt_tmp_136, in_2=_pt_tmp_82 )["out"] ) del _pt_tmp_183 _pt_tmp_180 = ( actx.np.where(_pt_tmp_89, _pt_tmp_181, 0) if actx.supports_nonscalar_broadcasting - else actx.call_loopy(_pt_t_unit_8, _in0=_pt_tmp_89, _in1=_pt_tmp_181)["out"] + else actx.call_loopy(_pt_t_unit_6, _in0=_pt_tmp_89, _in1=_pt_tmp_181)["out"] ) del _pt_tmp_182 _pt_tmp_180 = actx.tag_axis(0, (DiscretizationElementAxisTag(),), _pt_tmp_180) del _pt_tmp_181 _pt_tmp_180 = actx.tag_axis(1, (DiscretizationDOFAxisTag(),), _pt_tmp_180) - _pt_tmp_179 = 0j + _pt_tmp_180 - _pt_tmp_178 = 0j + _pt_tmp_179 + _pt_tmp_179 = 0.0 + _pt_tmp_180 + _pt_tmp_178 = 0.0 + _pt_tmp_179 del _pt_tmp_180 _pt_tmp_177 = _pt_tmp_178 + _pt_tmp_137 del _pt_tmp_179 @@ -849,105 +787,101 @@ def _rhs_inner( _pt_tmp_150 = actx.tag((FirstAxisIsElementsTag(),), _pt_tmp_150) del _pt_tmp_151 _pt_tmp_139 = _pt_tmp_140 - _pt_tmp_150 - _pt_tmp_138 = _pt_tmp_139 / np.complex128(1 + 0j) + _pt_tmp_138 = _pt_tmp_139 / np.float64(1.0) del _pt_tmp_140, _pt_tmp_150 _pt_tmp_204 = actx.einsum( "ij,ikl,jl->jk", _pt_tmp_146, _pt_data_0, _actx_in_1_4_0 ) del _pt_tmp_139 _pt_tmp_204 = actx.tag((FirstAxisIsElementsTag(),), _pt_tmp_204) - _pt_tmp_203 = (1 + 0j) * _pt_tmp_204 - _pt_tmp_202 = 0j + _pt_tmp_203 + _pt_tmp_203 = 1.0 * _pt_tmp_204 + _pt_tmp_202 = 0.0 + _pt_tmp_203 del _pt_tmp_204 _pt_tmp_206 = actx.einsum("ij,ikl,jl->jk", _pt_tmp_8, _pt_data_0, _actx_in_1_3_0) del _pt_tmp_203 _pt_tmp_206 = actx.tag((FirstAxisIsElementsTag(),), _pt_tmp_206) - _pt_tmp_205 = (-1 + 0j) * _pt_tmp_206 + _pt_tmp_205 = -1.0 * _pt_tmp_206 _pt_tmp_201 = _pt_tmp_202 + _pt_tmp_205 del _pt_tmp_206 - _pt_tmp_200 = 0j - _pt_tmp_201 + _pt_tmp_200 = 0.0 - _pt_tmp_201 del _pt_tmp_202, _pt_tmp_205 _pt_tmp_199 = -1 * _pt_tmp_200 del _pt_tmp_201 _pt_tmp_218 = ( _pt_tmp_47 * _pt_tmp_62 if actx.supports_nonscalar_broadcasting - else actx.call_loopy(_pt_t_unit_5, _in0=_pt_tmp_47, _in1=_pt_tmp_62)["out"] + else actx.call_loopy(_pt_t_unit_3, _in0=_pt_tmp_47, _in1=_pt_tmp_62)["out"] ) del _pt_tmp_200 - _pt_tmp_217 = 0j + _pt_tmp_218 + _pt_tmp_217 = 0.0 + _pt_tmp_218 del _pt_tmp_62 _pt_tmp_219 = ( _pt_tmp_54 * _pt_tmp_163 if actx.supports_nonscalar_broadcasting - else actx.call_loopy(_pt_t_unit_5, _in0=_pt_tmp_54, _in1=_pt_tmp_163)["out"] + else actx.call_loopy(_pt_t_unit_3, _in0=_pt_tmp_54, _in1=_pt_tmp_163)["out"] ) del _pt_tmp_218 _pt_tmp_216 = _pt_tmp_217 + _pt_tmp_219 del _pt_tmp_163 - _pt_tmp_215 = (-0.5 + 0j) * _pt_tmp_216 + _pt_tmp_215 = -0.5 * _pt_tmp_216 del _pt_tmp_217, _pt_tmp_219 _pt_tmp_214 = ( _pt_tmp_215[_pt_tmp_81, _pt_tmp_82] if actx.permits_advanced_indexing else actx.call_loopy( - _pt_t_unit_7, in_0=_pt_tmp_215, in_1=_pt_tmp_81, in_2=_pt_tmp_82 + _pt_t_unit_5, in_0=_pt_tmp_215, in_1=_pt_tmp_81, in_2=_pt_tmp_82 )["out"] ) del _pt_tmp_216 _pt_tmp_213 = ( actx.np.where(_pt_tmp_25, _pt_tmp_214, 0) if actx.supports_nonscalar_broadcasting - else actx.call_loopy(_pt_t_unit_8, _in0=_pt_tmp_25, _in1=_pt_tmp_214)["out"] + else actx.call_loopy(_pt_t_unit_6, _in0=_pt_tmp_25, _in1=_pt_tmp_214)["out"] ) del _pt_tmp_215 _pt_tmp_213 = actx.tag_axis(0, (DiscretizationElementAxisTag(),), _pt_tmp_213) del _pt_tmp_214 _pt_tmp_213 = actx.tag_axis(1, (DiscretizationDOFAxisTag(),), _pt_tmp_213) - _pt_tmp_212 = 0j + _pt_tmp_213 - _pt_tmp_211 = 0j + _pt_tmp_212 + _pt_tmp_212 = 0.0 + _pt_tmp_213 + _pt_tmp_211 = 0.0 + _pt_tmp_212 del _pt_tmp_213 _pt_tmp_230 = ( _pt_tmp_107 * _pt_tmp_120 if actx.supports_nonscalar_broadcasting - else actx.call_loopy(_pt_t_unit_12, _in0=_pt_tmp_107, _in1=_pt_tmp_120)[ - "out" - ] + else actx.call_loopy(_pt_t_unit_9, _in0=_pt_tmp_107, _in1=_pt_tmp_120)["out"] ) del _pt_tmp_212 - _pt_tmp_229 = 0j + _pt_tmp_230 + _pt_tmp_229 = 0.0 + _pt_tmp_230 del _pt_tmp_120 _pt_tmp_231 = ( _pt_tmp_113 * _pt_tmp_187 if actx.supports_nonscalar_broadcasting - else actx.call_loopy(_pt_t_unit_12, _in0=_pt_tmp_113, _in1=_pt_tmp_187)[ - "out" - ] + else actx.call_loopy(_pt_t_unit_9, _in0=_pt_tmp_113, _in1=_pt_tmp_187)["out"] ) del _pt_tmp_230 _pt_tmp_228 = _pt_tmp_229 + _pt_tmp_231 del _pt_tmp_187 - _pt_tmp_227 = (-0.5 + 0j) * _pt_tmp_228 + _pt_tmp_227 = -0.5 * _pt_tmp_228 del _pt_tmp_229, _pt_tmp_231 _pt_tmp_226 = ( _pt_tmp_227[_pt_tmp_136, _pt_tmp_82] if actx.permits_advanced_indexing else actx.call_loopy( - _pt_t_unit_13, in_0=_pt_tmp_227, in_1=_pt_tmp_136, in_2=_pt_tmp_82 + _pt_t_unit_10, in_0=_pt_tmp_227, in_1=_pt_tmp_136, in_2=_pt_tmp_82 )["out"] ) del _pt_tmp_228 _pt_tmp_225 = ( actx.np.where(_pt_tmp_89, _pt_tmp_226, 0) if actx.supports_nonscalar_broadcasting - else actx.call_loopy(_pt_t_unit_8, _in0=_pt_tmp_89, _in1=_pt_tmp_226)["out"] + else actx.call_loopy(_pt_t_unit_6, _in0=_pt_tmp_89, _in1=_pt_tmp_226)["out"] ) del _pt_tmp_227 _pt_tmp_225 = actx.tag_axis(0, (DiscretizationElementAxisTag(),), _pt_tmp_225) del _pt_tmp_226 _pt_tmp_225 = actx.tag_axis(1, (DiscretizationDOFAxisTag(),), _pt_tmp_225) - _pt_tmp_224 = 0j + _pt_tmp_225 - _pt_tmp_223 = 0j + _pt_tmp_224 + _pt_tmp_224 = 0.0 + _pt_tmp_225 + _pt_tmp_223 = 0.0 + _pt_tmp_224 del _pt_tmp_225 _pt_tmp_222 = _pt_tmp_223 + _pt_tmp_137 del _pt_tmp_224 @@ -968,185 +902,177 @@ def _rhs_inner( _pt_tmp_207 = actx.tag((FirstAxisIsElementsTag(),), _pt_tmp_207) del _pt_tmp_208 _pt_tmp_198 = _pt_tmp_199 - _pt_tmp_207 - _pt_tmp_197 = _pt_tmp_198 / np.complex128(1 + 0j) + _pt_tmp_197 = _pt_tmp_198 / np.float64(1.0) del _pt_tmp_199, _pt_tmp_207 _pt_tmp_238 = actx.einsum("ij,ikl,jl->jk", _pt_tmp_8, _pt_data_0, _actx_in_1_2_0) del _pt_tmp_198 _pt_tmp_238 = actx.tag((FirstAxisIsElementsTag(),), _pt_tmp_238) - _pt_tmp_237 = (1 + 0j) * _pt_tmp_238 - _pt_tmp_236 = 0j + _pt_tmp_237 + _pt_tmp_237 = 1.0 * _pt_tmp_238 + _pt_tmp_236 = 0.0 + _pt_tmp_237 del _pt_tmp_238 _pt_tmp_240 = actx.einsum( "ij,ikl,jl->jk", _pt_tmp_12, _pt_data_0, _actx_in_1_1_0 ) del _pt_tmp_237 _pt_tmp_240 = actx.tag((FirstAxisIsElementsTag(),), _pt_tmp_240) - _pt_tmp_239 = (-1 + 0j) * _pt_tmp_240 + _pt_tmp_239 = -1.0 * _pt_tmp_240 _pt_tmp_235 = _pt_tmp_236 + _pt_tmp_239 del _pt_tmp_240 _pt_tmp_234 = -1 * _pt_tmp_235 del _pt_tmp_236, _pt_tmp_239 - _pt_tmp_254 = (1 + 0j) * _pt_tmp_74 + _pt_tmp_254 = 1.0 * _pt_tmp_74 del _pt_tmp_235 _pt_tmp_258 = ( _pt_tmp_47 * _pt_tmp_64 if actx.supports_nonscalar_broadcasting - else actx.call_loopy(_pt_t_unit_5, _in0=_pt_tmp_47, _in1=_pt_tmp_64)["out"] + else actx.call_loopy(_pt_t_unit_3, _in0=_pt_tmp_47, _in1=_pt_tmp_64)["out"] ) del _pt_tmp_74 - _pt_tmp_257 = 0j + _pt_tmp_258 + _pt_tmp_257 = 0.0 + _pt_tmp_258 _pt_tmp_259 = ( _pt_tmp_54 * _pt_tmp_165 if actx.supports_nonscalar_broadcasting - else actx.call_loopy(_pt_t_unit_5, _in0=_pt_tmp_54, _in1=_pt_tmp_165)["out"] + else actx.call_loopy(_pt_t_unit_3, _in0=_pt_tmp_54, _in1=_pt_tmp_165)["out"] ) del _pt_tmp_258 _pt_tmp_256 = _pt_tmp_257 + _pt_tmp_259 - _pt_tmp_255 = (0.5 + 0j) * _pt_tmp_256 + _pt_tmp_255 = 0.5 * _pt_tmp_256 del _pt_tmp_257, _pt_tmp_259 _pt_tmp_253 = _pt_tmp_254 + _pt_tmp_255 del _pt_tmp_256 _pt_tmp_252 = ( _pt_tmp_31 * _pt_tmp_253 if actx.supports_nonscalar_broadcasting - else actx.call_loopy(_pt_t_unit_5, _in0=_pt_tmp_31, _in1=_pt_tmp_253)["out"] + else actx.call_loopy(_pt_t_unit_3, _in0=_pt_tmp_31, _in1=_pt_tmp_253)["out"] ) del _pt_tmp_254, _pt_tmp_255 - _pt_tmp_251 = 0j + _pt_tmp_252 - _pt_tmp_262 = (1 + 0j) * _pt_tmp_48 + _pt_tmp_251 = 0.0 + _pt_tmp_252 + _pt_tmp_262 = 1.0 * _pt_tmp_48 del _pt_tmp_252 _pt_tmp_266 = ( _pt_tmp_73 * _pt_tmp_34 if actx.supports_nonscalar_broadcasting - else actx.call_loopy(_pt_t_unit_14, _in0=_pt_tmp_73, _in1=_pt_tmp_34)["out"] + else actx.call_loopy(_pt_t_unit_3, _in0=_pt_tmp_73, _in1=_pt_tmp_34)["out"] ) del _pt_tmp_48 _pt_tmp_265 = 0.0 + _pt_tmp_266 _pt_tmp_267 = ( _pt_tmp_80 * _pt_tmp_165 if actx.supports_nonscalar_broadcasting - else actx.call_loopy(_pt_t_unit_5, _in0=_pt_tmp_80, _in1=_pt_tmp_165)["out"] + else actx.call_loopy(_pt_t_unit_3, _in0=_pt_tmp_80, _in1=_pt_tmp_165)["out"] ) del _pt_tmp_266 _pt_tmp_264 = _pt_tmp_265 + _pt_tmp_267 del _pt_tmp_165 - _pt_tmp_263 = (0.5 + 0j) * _pt_tmp_264 + _pt_tmp_263 = 0.5 * _pt_tmp_264 del _pt_tmp_265, _pt_tmp_267 _pt_tmp_261 = _pt_tmp_262 + _pt_tmp_263 del _pt_tmp_264 _pt_tmp_260 = ( _pt_tmp_61 * _pt_tmp_261 if actx.supports_nonscalar_broadcasting - else actx.call_loopy(_pt_t_unit_5, _in0=_pt_tmp_61, _in1=_pt_tmp_261)["out"] + else actx.call_loopy(_pt_t_unit_3, _in0=_pt_tmp_61, _in1=_pt_tmp_261)["out"] ) del _pt_tmp_262, _pt_tmp_263 _pt_tmp_250 = _pt_tmp_251 + _pt_tmp_260 - _pt_tmp_249 = (0.5 + 0j) * _pt_tmp_250 + _pt_tmp_249 = 0.5 * _pt_tmp_250 del _pt_tmp_251, _pt_tmp_260 _pt_tmp_248 = ( _pt_tmp_249[_pt_tmp_81, _pt_tmp_82] if actx.permits_advanced_indexing else actx.call_loopy( - _pt_t_unit_7, in_0=_pt_tmp_249, in_1=_pt_tmp_81, in_2=_pt_tmp_82 + _pt_t_unit_5, in_0=_pt_tmp_249, in_1=_pt_tmp_81, in_2=_pt_tmp_82 )["out"] ) del _pt_tmp_250 _pt_tmp_247 = ( actx.np.where(_pt_tmp_25, _pt_tmp_248, 0) if actx.supports_nonscalar_broadcasting - else actx.call_loopy(_pt_t_unit_8, _in0=_pt_tmp_25, _in1=_pt_tmp_248)["out"] + else actx.call_loopy(_pt_t_unit_6, _in0=_pt_tmp_25, _in1=_pt_tmp_248)["out"] ) del _pt_tmp_249 _pt_tmp_247 = actx.tag_axis(0, (DiscretizationElementAxisTag(),), _pt_tmp_247) del _pt_tmp_248 _pt_tmp_247 = actx.tag_axis(1, (DiscretizationDOFAxisTag(),), _pt_tmp_247) - _pt_tmp_246 = 0j + _pt_tmp_247 - _pt_tmp_245 = 0j + _pt_tmp_246 + _pt_tmp_246 = 0.0 + _pt_tmp_247 + _pt_tmp_245 = 0.0 + _pt_tmp_246 del _pt_tmp_247 - _pt_tmp_280 = (1 + 0j) * _pt_tmp_130 + _pt_tmp_280 = 1.0 * _pt_tmp_130 del _pt_tmp_246 _pt_tmp_284 = ( _pt_tmp_107 * _pt_tmp_122 if actx.supports_nonscalar_broadcasting - else actx.call_loopy(_pt_t_unit_12, _in0=_pt_tmp_107, _in1=_pt_tmp_122)[ - "out" - ] + else actx.call_loopy(_pt_t_unit_9, _in0=_pt_tmp_107, _in1=_pt_tmp_122)["out"] ) del _pt_tmp_130 - _pt_tmp_283 = 0j + _pt_tmp_284 + _pt_tmp_283 = 0.0 + _pt_tmp_284 _pt_tmp_285 = ( _pt_tmp_113 * _pt_tmp_189 if actx.supports_nonscalar_broadcasting - else actx.call_loopy(_pt_t_unit_12, _in0=_pt_tmp_113, _in1=_pt_tmp_189)[ - "out" - ] + else actx.call_loopy(_pt_t_unit_9, _in0=_pt_tmp_113, _in1=_pt_tmp_189)["out"] ) del _pt_tmp_284 _pt_tmp_282 = _pt_tmp_283 + _pt_tmp_285 - _pt_tmp_281 = (0.5 + 0j) * _pt_tmp_282 + _pt_tmp_281 = 0.5 * _pt_tmp_282 del _pt_tmp_283, _pt_tmp_285 _pt_tmp_279 = _pt_tmp_280 + _pt_tmp_281 del _pt_tmp_282 _pt_tmp_278 = ( _pt_tmp_95 * _pt_tmp_279 if actx.supports_nonscalar_broadcasting - else actx.call_loopy(_pt_t_unit_12, _in0=_pt_tmp_95, _in1=_pt_tmp_279)["out"] + else actx.call_loopy(_pt_t_unit_9, _in0=_pt_tmp_95, _in1=_pt_tmp_279)["out"] ) del _pt_tmp_280, _pt_tmp_281 - _pt_tmp_277 = 0j + _pt_tmp_278 - _pt_tmp_288 = (1 + 0j) * _pt_tmp_108 + _pt_tmp_277 = 0.0 + _pt_tmp_278 + _pt_tmp_288 = 1.0 * _pt_tmp_108 del _pt_tmp_278 _pt_tmp_292 = ( _pt_tmp_129 * _pt_tmp_98 if actx.supports_nonscalar_broadcasting - else actx.call_loopy(_pt_t_unit_15, _in0=_pt_tmp_129, _in1=_pt_tmp_98)["out"] + else actx.call_loopy(_pt_t_unit_9, _in0=_pt_tmp_129, _in1=_pt_tmp_98)["out"] ) del _pt_tmp_108 _pt_tmp_291 = 0.0 + _pt_tmp_292 _pt_tmp_293 = ( _pt_tmp_135 * _pt_tmp_189 if actx.supports_nonscalar_broadcasting - else actx.call_loopy(_pt_t_unit_12, _in0=_pt_tmp_135, _in1=_pt_tmp_189)[ - "out" - ] + else actx.call_loopy(_pt_t_unit_9, _in0=_pt_tmp_135, _in1=_pt_tmp_189)["out"] ) del _pt_tmp_292 _pt_tmp_290 = _pt_tmp_291 + _pt_tmp_293 del _pt_tmp_189 - _pt_tmp_289 = (0.5 + 0j) * _pt_tmp_290 + _pt_tmp_289 = 0.5 * _pt_tmp_290 del _pt_tmp_291, _pt_tmp_293 _pt_tmp_287 = _pt_tmp_288 + _pt_tmp_289 del _pt_tmp_290 _pt_tmp_286 = ( _pt_tmp_119 * _pt_tmp_287 if actx.supports_nonscalar_broadcasting - else actx.call_loopy(_pt_t_unit_12, _in0=_pt_tmp_119, _in1=_pt_tmp_287)[ - "out" - ] + else actx.call_loopy(_pt_t_unit_9, _in0=_pt_tmp_119, _in1=_pt_tmp_287)["out"] ) del _pt_tmp_288, _pt_tmp_289 _pt_tmp_276 = _pt_tmp_277 + _pt_tmp_286 - _pt_tmp_275 = (0.5 + 0j) * _pt_tmp_276 + _pt_tmp_275 = 0.5 * _pt_tmp_276 del _pt_tmp_277, _pt_tmp_286 _pt_tmp_274 = ( _pt_tmp_275[_pt_tmp_136, _pt_tmp_82] if actx.permits_advanced_indexing else actx.call_loopy( - _pt_t_unit_13, in_0=_pt_tmp_275, in_1=_pt_tmp_136, in_2=_pt_tmp_82 + _pt_t_unit_10, in_0=_pt_tmp_275, in_1=_pt_tmp_136, in_2=_pt_tmp_82 )["out"] ) del _pt_tmp_276 _pt_tmp_273 = ( actx.np.where(_pt_tmp_89, _pt_tmp_274, 0) if actx.supports_nonscalar_broadcasting - else actx.call_loopy(_pt_t_unit_8, _in0=_pt_tmp_89, _in1=_pt_tmp_274)["out"] + else actx.call_loopy(_pt_t_unit_6, _in0=_pt_tmp_89, _in1=_pt_tmp_274)["out"] ) del _pt_tmp_275 _pt_tmp_273 = actx.tag_axis(0, (DiscretizationElementAxisTag(),), _pt_tmp_273) del _pt_tmp_274 _pt_tmp_273 = actx.tag_axis(1, (DiscretizationDOFAxisTag(),), _pt_tmp_273) - _pt_tmp_272 = 0j + _pt_tmp_273 - _pt_tmp_271 = 0j + _pt_tmp_272 + _pt_tmp_272 = 0.0 + _pt_tmp_273 + _pt_tmp_271 = 0.0 + _pt_tmp_272 del _pt_tmp_273 _pt_tmp_270 = _pt_tmp_271 + _pt_tmp_137 del _pt_tmp_272 @@ -1167,15 +1093,15 @@ def _rhs_inner( _pt_tmp_241 = actx.tag((FirstAxisIsElementsTag(),), _pt_tmp_241) del _pt_tmp_242 _pt_tmp_233 = _pt_tmp_234 - _pt_tmp_241 - _pt_tmp_232 = _pt_tmp_233 / np.complex128(1 + 0j) + _pt_tmp_232 = _pt_tmp_233 / np.float64(1.0) del _pt_tmp_234, _pt_tmp_241 _pt_tmp_300 = actx.einsum( "ij,ikl,jl->jk", _pt_tmp_146, _pt_data_0, _actx_in_1_2_0 ) del _pt_tmp_233 _pt_tmp_300 = actx.tag((FirstAxisIsElementsTag(),), _pt_tmp_300) - _pt_tmp_299 = (-1 + 0j) * _pt_tmp_300 - _pt_tmp_298 = 0j + _pt_tmp_299 + _pt_tmp_299 = -1.0 * _pt_tmp_300 + _pt_tmp_298 = 0.0 + _pt_tmp_299 del _pt_tmp_300 _pt_tmp_302 = actx.einsum( "ij,ikl,jl->jk", _pt_tmp_12, _pt_data_0, _actx_in_1_0_0 @@ -1183,7 +1109,7 @@ def _rhs_inner( del _pt_tmp_299 _pt_tmp_302 = actx.tag((FirstAxisIsElementsTag(),), _pt_tmp_302) del _pt_tmp_12 - _pt_tmp_301 = (1 + 0j) * _pt_tmp_302 + _pt_tmp_301 = 1.0 * _pt_tmp_302 _pt_tmp_297 = _pt_tmp_298 + _pt_tmp_301 del _pt_tmp_302 _pt_tmp_296 = -1 * _pt_tmp_297 @@ -1191,17 +1117,17 @@ def _rhs_inner( _pt_tmp_314 = ( _pt_tmp_73 * _pt_tmp_253 if actx.supports_nonscalar_broadcasting - else actx.call_loopy(_pt_t_unit_5, _in0=_pt_tmp_73, _in1=_pt_tmp_253)["out"] + else actx.call_loopy(_pt_t_unit_3, _in0=_pt_tmp_73, _in1=_pt_tmp_253)["out"] ) del _pt_tmp_297 - _pt_tmp_313 = 0j + _pt_tmp_314 + _pt_tmp_313 = 0.0 + _pt_tmp_314 del _pt_tmp_253, _pt_tmp_73 - _pt_tmp_317 = (1 + 0j) * _pt_tmp_55 + _pt_tmp_317 = 1.0 * _pt_tmp_55 del _pt_tmp_314 _pt_tmp_321 = ( _pt_tmp_31 * _pt_tmp_34 if actx.supports_nonscalar_broadcasting - else actx.call_loopy(_pt_t_unit_14, _in0=_pt_tmp_31, _in1=_pt_tmp_34)["out"] + else actx.call_loopy(_pt_t_unit_3, _in0=_pt_tmp_31, _in1=_pt_tmp_34)["out"] ) del _pt_tmp_55 _pt_tmp_320 = 0.0 + _pt_tmp_321 @@ -1209,61 +1135,59 @@ def _rhs_inner( _pt_tmp_322 = ( _pt_tmp_61 * _pt_tmp_64 if actx.supports_nonscalar_broadcasting - else actx.call_loopy(_pt_t_unit_5, _in0=_pt_tmp_61, _in1=_pt_tmp_64)["out"] + else actx.call_loopy(_pt_t_unit_3, _in0=_pt_tmp_61, _in1=_pt_tmp_64)["out"] ) del _pt_tmp_321 _pt_tmp_319 = _pt_tmp_320 + _pt_tmp_322 del _pt_tmp_61, _pt_tmp_64 - _pt_tmp_318 = (0.5 + 0j) * _pt_tmp_319 + _pt_tmp_318 = 0.5 * _pt_tmp_319 del _pt_tmp_320, _pt_tmp_322 _pt_tmp_316 = _pt_tmp_317 + _pt_tmp_318 del _pt_tmp_319 _pt_tmp_315 = ( _pt_tmp_80 * _pt_tmp_316 if actx.supports_nonscalar_broadcasting - else actx.call_loopy(_pt_t_unit_5, _in0=_pt_tmp_80, _in1=_pt_tmp_316)["out"] + else actx.call_loopy(_pt_t_unit_3, _in0=_pt_tmp_80, _in1=_pt_tmp_316)["out"] ) del _pt_tmp_317, _pt_tmp_318 _pt_tmp_312 = _pt_tmp_313 + _pt_tmp_315 del _pt_tmp_80 - _pt_tmp_311 = (0.5 + 0j) * _pt_tmp_312 + _pt_tmp_311 = 0.5 * _pt_tmp_312 del _pt_tmp_313, _pt_tmp_315 _pt_tmp_310 = ( _pt_tmp_311[_pt_tmp_81, _pt_tmp_82] if actx.permits_advanced_indexing else actx.call_loopy( - _pt_t_unit_7, in_0=_pt_tmp_311, in_1=_pt_tmp_81, in_2=_pt_tmp_82 + _pt_t_unit_5, in_0=_pt_tmp_311, in_1=_pt_tmp_81, in_2=_pt_tmp_82 )["out"] ) del _pt_tmp_312 _pt_tmp_309 = ( actx.np.where(_pt_tmp_25, _pt_tmp_310, 0) if actx.supports_nonscalar_broadcasting - else actx.call_loopy(_pt_t_unit_8, _in0=_pt_tmp_25, _in1=_pt_tmp_310)["out"] + else actx.call_loopy(_pt_t_unit_6, _in0=_pt_tmp_25, _in1=_pt_tmp_310)["out"] ) del _pt_tmp_311 _pt_tmp_309 = actx.tag_axis(0, (DiscretizationElementAxisTag(),), _pt_tmp_309) del _pt_tmp_310 _pt_tmp_309 = actx.tag_axis(1, (DiscretizationDOFAxisTag(),), _pt_tmp_309) - _pt_tmp_308 = 0j + _pt_tmp_309 - _pt_tmp_307 = 0j + _pt_tmp_308 + _pt_tmp_308 = 0.0 + _pt_tmp_309 + _pt_tmp_307 = 0.0 + _pt_tmp_308 del _pt_tmp_309 _pt_tmp_333 = ( _pt_tmp_129 * _pt_tmp_279 if actx.supports_nonscalar_broadcasting - else actx.call_loopy(_pt_t_unit_12, _in0=_pt_tmp_129, _in1=_pt_tmp_279)[ - "out" - ] + else actx.call_loopy(_pt_t_unit_9, _in0=_pt_tmp_129, _in1=_pt_tmp_279)["out"] ) del _pt_tmp_308 - _pt_tmp_332 = 0j + _pt_tmp_333 + _pt_tmp_332 = 0.0 + _pt_tmp_333 del _pt_tmp_129, _pt_tmp_279 - _pt_tmp_336 = (1 + 0j) * _pt_tmp_114 + _pt_tmp_336 = 1.0 * _pt_tmp_114 del _pt_tmp_333 _pt_tmp_340 = ( _pt_tmp_95 * _pt_tmp_98 if actx.supports_nonscalar_broadcasting - else actx.call_loopy(_pt_t_unit_15, _in0=_pt_tmp_95, _in1=_pt_tmp_98)["out"] + else actx.call_loopy(_pt_t_unit_9, _in0=_pt_tmp_95, _in1=_pt_tmp_98)["out"] ) del _pt_tmp_114 _pt_tmp_339 = 0.0 + _pt_tmp_340 @@ -1271,48 +1195,44 @@ def _rhs_inner( _pt_tmp_341 = ( _pt_tmp_119 * _pt_tmp_122 if actx.supports_nonscalar_broadcasting - else actx.call_loopy(_pt_t_unit_12, _in0=_pt_tmp_119, _in1=_pt_tmp_122)[ - "out" - ] + else actx.call_loopy(_pt_t_unit_9, _in0=_pt_tmp_119, _in1=_pt_tmp_122)["out"] ) del _pt_tmp_340 _pt_tmp_338 = _pt_tmp_339 + _pt_tmp_341 del _pt_tmp_119, _pt_tmp_122 - _pt_tmp_337 = (0.5 + 0j) * _pt_tmp_338 + _pt_tmp_337 = 0.5 * _pt_tmp_338 del _pt_tmp_339, _pt_tmp_341 _pt_tmp_335 = _pt_tmp_336 + _pt_tmp_337 del _pt_tmp_338 _pt_tmp_334 = ( _pt_tmp_135 * _pt_tmp_335 if actx.supports_nonscalar_broadcasting - else actx.call_loopy(_pt_t_unit_12, _in0=_pt_tmp_135, _in1=_pt_tmp_335)[ - "out" - ] + else actx.call_loopy(_pt_t_unit_9, _in0=_pt_tmp_135, _in1=_pt_tmp_335)["out"] ) del _pt_tmp_336, _pt_tmp_337 _pt_tmp_331 = _pt_tmp_332 + _pt_tmp_334 del _pt_tmp_135 - _pt_tmp_330 = (0.5 + 0j) * _pt_tmp_331 + _pt_tmp_330 = 0.5 * _pt_tmp_331 del _pt_tmp_332, _pt_tmp_334 _pt_tmp_329 = ( _pt_tmp_330[_pt_tmp_136, _pt_tmp_82] if actx.permits_advanced_indexing else actx.call_loopy( - _pt_t_unit_13, in_0=_pt_tmp_330, in_1=_pt_tmp_136, in_2=_pt_tmp_82 + _pt_t_unit_10, in_0=_pt_tmp_330, in_1=_pt_tmp_136, in_2=_pt_tmp_82 )["out"] ) del _pt_tmp_331 _pt_tmp_328 = ( actx.np.where(_pt_tmp_89, _pt_tmp_329, 0) if actx.supports_nonscalar_broadcasting - else actx.call_loopy(_pt_t_unit_8, _in0=_pt_tmp_89, _in1=_pt_tmp_329)["out"] + else actx.call_loopy(_pt_t_unit_6, _in0=_pt_tmp_89, _in1=_pt_tmp_329)["out"] ) del _pt_tmp_330 _pt_tmp_328 = actx.tag_axis(0, (DiscretizationElementAxisTag(),), _pt_tmp_328) del _pt_tmp_329 _pt_tmp_328 = actx.tag_axis(1, (DiscretizationDOFAxisTag(),), _pt_tmp_328) - _pt_tmp_327 = 0j + _pt_tmp_328 - _pt_tmp_326 = 0j + _pt_tmp_327 + _pt_tmp_327 = 0.0 + _pt_tmp_328 + _pt_tmp_326 = 0.0 + _pt_tmp_327 del _pt_tmp_328 _pt_tmp_325 = _pt_tmp_326 + _pt_tmp_137 del _pt_tmp_327 @@ -1333,7 +1253,7 @@ def _rhs_inner( _pt_tmp_303 = actx.tag((FirstAxisIsElementsTag(),), _pt_tmp_303) del _pt_tmp_304 _pt_tmp_295 = _pt_tmp_296 - _pt_tmp_303 - _pt_tmp_294 = _pt_tmp_295 / np.complex128(1 + 0j) + _pt_tmp_294 = _pt_tmp_295 / np.float64(1.0) del _pt_tmp_296, _pt_tmp_303 _pt_tmp_348 = actx.einsum( "ij,ikl,jl->jk", _pt_tmp_146, _pt_data_0, _actx_in_1_1_0 @@ -1341,14 +1261,14 @@ def _rhs_inner( del _pt_tmp_295 _pt_tmp_348 = actx.tag((FirstAxisIsElementsTag(),), _pt_tmp_348) del _pt_tmp_146 - _pt_tmp_347 = (1 + 0j) * _pt_tmp_348 - _pt_tmp_346 = 0j + _pt_tmp_347 + _pt_tmp_347 = 1.0 * _pt_tmp_348 + _pt_tmp_346 = 0.0 + _pt_tmp_347 del _pt_tmp_348 _pt_tmp_350 = actx.einsum("ij,ikl,jl->jk", _pt_tmp_8, _pt_data_0, _actx_in_1_0_0) del _pt_tmp_347 _pt_tmp_350 = actx.tag((FirstAxisIsElementsTag(),), _pt_tmp_350) del _pt_tmp_8 - _pt_tmp_349 = (-1 + 0j) * _pt_tmp_350 + _pt_tmp_349 = -1.0 * _pt_tmp_350 _pt_tmp_345 = _pt_tmp_346 + _pt_tmp_349 del _pt_tmp_350 _pt_tmp_344 = -1 * _pt_tmp_345 @@ -1356,82 +1276,78 @@ def _rhs_inner( _pt_tmp_362 = ( _pt_tmp_47 * _pt_tmp_261 if actx.supports_nonscalar_broadcasting - else actx.call_loopy(_pt_t_unit_5, _in0=_pt_tmp_47, _in1=_pt_tmp_261)["out"] + else actx.call_loopy(_pt_t_unit_3, _in0=_pt_tmp_47, _in1=_pt_tmp_261)["out"] ) del _pt_tmp_345 - _pt_tmp_361 = 0j + _pt_tmp_362 + _pt_tmp_361 = 0.0 + _pt_tmp_362 del _pt_tmp_261, _pt_tmp_47 _pt_tmp_363 = ( _pt_tmp_54 * _pt_tmp_316 if actx.supports_nonscalar_broadcasting - else actx.call_loopy(_pt_t_unit_5, _in0=_pt_tmp_54, _in1=_pt_tmp_316)["out"] + else actx.call_loopy(_pt_t_unit_3, _in0=_pt_tmp_54, _in1=_pt_tmp_316)["out"] ) del _pt_tmp_362 _pt_tmp_360 = _pt_tmp_361 + _pt_tmp_363 del _pt_tmp_316, _pt_tmp_54 - _pt_tmp_359 = (0.5 + 0j) * _pt_tmp_360 + _pt_tmp_359 = 0.5 * _pt_tmp_360 del _pt_tmp_361, _pt_tmp_363 _pt_tmp_358 = ( _pt_tmp_359[_pt_tmp_81, _pt_tmp_82] if actx.permits_advanced_indexing else actx.call_loopy( - _pt_t_unit_7, in_0=_pt_tmp_359, in_1=_pt_tmp_81, in_2=_pt_tmp_82 + _pt_t_unit_5, in_0=_pt_tmp_359, in_1=_pt_tmp_81, in_2=_pt_tmp_82 )["out"] ) del _pt_tmp_360 _pt_tmp_357 = ( actx.np.where(_pt_tmp_25, _pt_tmp_358, 0) if actx.supports_nonscalar_broadcasting - else actx.call_loopy(_pt_t_unit_8, _in0=_pt_tmp_25, _in1=_pt_tmp_358)["out"] + else actx.call_loopy(_pt_t_unit_6, _in0=_pt_tmp_25, _in1=_pt_tmp_358)["out"] ) del _pt_tmp_359, _pt_tmp_81 _pt_tmp_357 = actx.tag_axis(0, (DiscretizationElementAxisTag(),), _pt_tmp_357) del _pt_tmp_25, _pt_tmp_358 _pt_tmp_357 = actx.tag_axis(1, (DiscretizationDOFAxisTag(),), _pt_tmp_357) - _pt_tmp_356 = 0j + _pt_tmp_357 - _pt_tmp_355 = 0j + _pt_tmp_356 + _pt_tmp_356 = 0.0 + _pt_tmp_357 + _pt_tmp_355 = 0.0 + _pt_tmp_356 del _pt_tmp_357 _pt_tmp_374 = ( _pt_tmp_107 * _pt_tmp_287 if actx.supports_nonscalar_broadcasting - else actx.call_loopy(_pt_t_unit_12, _in0=_pt_tmp_107, _in1=_pt_tmp_287)[ - "out" - ] + else actx.call_loopy(_pt_t_unit_9, _in0=_pt_tmp_107, _in1=_pt_tmp_287)["out"] ) del _pt_tmp_356 - _pt_tmp_373 = 0j + _pt_tmp_374 + _pt_tmp_373 = 0.0 + _pt_tmp_374 del _pt_tmp_107, _pt_tmp_287 _pt_tmp_375 = ( _pt_tmp_113 * _pt_tmp_335 if actx.supports_nonscalar_broadcasting - else actx.call_loopy(_pt_t_unit_12, _in0=_pt_tmp_113, _in1=_pt_tmp_335)[ - "out" - ] + else actx.call_loopy(_pt_t_unit_9, _in0=_pt_tmp_113, _in1=_pt_tmp_335)["out"] ) del _pt_tmp_374 _pt_tmp_372 = _pt_tmp_373 + _pt_tmp_375 del _pt_tmp_113, _pt_tmp_335 - _pt_tmp_371 = (0.5 + 0j) * _pt_tmp_372 + _pt_tmp_371 = 0.5 * _pt_tmp_372 del _pt_tmp_373, _pt_tmp_375 _pt_tmp_370 = ( _pt_tmp_371[_pt_tmp_136, _pt_tmp_82] if actx.permits_advanced_indexing else actx.call_loopy( - _pt_t_unit_13, in_0=_pt_tmp_371, in_1=_pt_tmp_136, in_2=_pt_tmp_82 + _pt_t_unit_10, in_0=_pt_tmp_371, in_1=_pt_tmp_136, in_2=_pt_tmp_82 )["out"] ) del _pt_tmp_372 _pt_tmp_369 = ( actx.np.where(_pt_tmp_89, _pt_tmp_370, 0) if actx.supports_nonscalar_broadcasting - else actx.call_loopy(_pt_t_unit_8, _in0=_pt_tmp_89, _in1=_pt_tmp_370)["out"] + else actx.call_loopy(_pt_t_unit_6, _in0=_pt_tmp_89, _in1=_pt_tmp_370)["out"] ) del _pt_tmp_136, _pt_tmp_371, _pt_tmp_82 _pt_tmp_369 = actx.tag_axis(0, (DiscretizationElementAxisTag(),), _pt_tmp_369) del _pt_tmp_370, _pt_tmp_89 _pt_tmp_369 = actx.tag_axis(1, (DiscretizationDOFAxisTag(),), _pt_tmp_369) - _pt_tmp_368 = 0j + _pt_tmp_369 - _pt_tmp_367 = 0j + _pt_tmp_368 + _pt_tmp_368 = 0.0 + _pt_tmp_369 + _pt_tmp_367 = 0.0 + _pt_tmp_368 del _pt_tmp_369 _pt_tmp_366 = _pt_tmp_367 + _pt_tmp_137 del _pt_tmp_368 @@ -1452,7 +1368,7 @@ def _rhs_inner( _pt_tmp_351 = actx.tag((FirstAxisIsElementsTag(),), _pt_tmp_351) del _pt_tmp_15, _pt_tmp_352 _pt_tmp_343 = _pt_tmp_344 - _pt_tmp_351 - _pt_tmp_342 = _pt_tmp_343 / np.complex128(1 + 0j) + _pt_tmp_342 = _pt_tmp_343 / np.float64(1.0) del _pt_tmp_344, _pt_tmp_351 _pt_tmp = make_obj_array( [_pt_tmp_0, _pt_tmp_138, _pt_tmp_197, _pt_tmp_232, _pt_tmp_294, _pt_tmp_342] diff --git a/actx_dgfem_suite/suite/tiny_maxwell_3D_P4/ref_input_args.pkl.dvc b/actx_dgfem_suite/suite/tiny_maxwell_3D_P4/ref_input_args.pkl.dvc index 557cd39..bb1ae53 100644 --- a/actx_dgfem_suite/suite/tiny_maxwell_3D_P4/ref_input_args.pkl.dvc +++ b/actx_dgfem_suite/suite/tiny_maxwell_3D_P4/ref_input_args.pkl.dvc @@ -1,5 +1,5 @@ outs: -- md5: 2ea88ddeb180f3dbde6281c77a5bf594 - size: 148570 +- md5: ecc963520bcd7b5d2964bda925f7ec77 + size: 81326 hash: md5 path: ref_input_args.pkl diff --git a/actx_dgfem_suite/suite/tiny_maxwell_3D_P4/ref_outputs.pkl.dvc b/actx_dgfem_suite/suite/tiny_maxwell_3D_P4/ref_outputs.pkl.dvc index 4c1dc9e..c3df12a 100644 --- a/actx_dgfem_suite/suite/tiny_maxwell_3D_P4/ref_outputs.pkl.dvc +++ b/actx_dgfem_suite/suite/tiny_maxwell_3D_P4/ref_outputs.pkl.dvc @@ -1,5 +1,5 @@ outs: -- md5: fabe694095004ad66814cebc46cbee47 - size: 161940 +- md5: 9b0e4f5f0a3dbb3c8fc4ccfc0f12f629 + size: 81290 hash: md5 path: ref_outputs.pkl diff --git a/actx_dgfem_suite/suite_generators/maxwell.py b/actx_dgfem_suite/suite_generators/maxwell.py index c20b0c0..209a69b 100644 --- a/actx_dgfem_suite/suite_generators/maxwell.py +++ b/actx_dgfem_suite/suite_generators/maxwell.py @@ -70,4 +70,6 @@ def rhs(t: float, w: Vector) -> Vector: compiled_rhs = actx.compile(rhs) fields = actx.freeze_thaw(fields) + # Hack to forces real only fields. + fields = actx.np.real(fields) compiled_rhs(np.float64(0.5), fields)