def test_transpose2(x, y, axis1, axis2): perm = (scalar_cast(axis1, u64), scalar_cast(axis2, u64)) xt = transpose(x, perm) yt = transpose(y, perm) d = dot(xt, yt) sm = array_reduce(scalar_add, d, ()) return array_to_scalar(sm)
def test_scalar_cast(): assert isinstance(scalar_cast(1.5, type_to_abstract(i64)), np.int64) assert isinstance(scalar_cast(1.5, type_to_abstract(f16)), np.float16)
def test_cast_to_dtype(arr, x): return scalar_cast(x, dtype(arr))