// Copyright 2022 The Tint Authors. // // Licensed under the Apache License, Version 2.0 (the "License"); // you may not use this file except in compliance with the License. // You may obtain a copy of the License at // // http://www.apache.org/licenses/LICENSE-2.0 // // Unless required by applicable law or agreed to in writing, software // distributed under the License is distributed on an "AS IS" BASIS, // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. // See the License for the specific language governing permissions and // limitations under the License. #include "src/tint/resolver/const_eval_test.h" using namespace tint::number_suffixes; // NOLINT namespace tint::resolver { namespace { TEST_F(ResolverConstEvalTest, Scalar_AFloat) { auto* expr = Expr(99.0_a); auto* a = Const("a", expr); WrapInFunction(a); EXPECT_TRUE(r()->Resolve()) << r()->error(); auto* sem = Sem().Get(expr); ASSERT_NE(sem, nullptr); EXPECT_TRUE(sem->Type()->Is()); EXPECT_TYPE(sem->ConstantValue()->Type(), sem->Type()); EXPECT_TRUE(sem->ConstantValue()->AllEqual()); EXPECT_FALSE(sem->ConstantValue()->AnyZero()); EXPECT_FALSE(sem->ConstantValue()->AllZero()); EXPECT_EQ(sem->ConstantValue()->ValueAs(), 99.0f); } TEST_F(ResolverConstEvalTest, Scalar_AInt) { auto* expr = Expr(99_a); auto* a = Const("a", expr); WrapInFunction(a); EXPECT_TRUE(r()->Resolve()) << r()->error(); auto* sem = Sem().Get(expr); ASSERT_NE(sem, nullptr); EXPECT_TRUE(sem->Type()->Is()); EXPECT_TYPE(sem->ConstantValue()->Type(), sem->Type()); EXPECT_TRUE(sem->ConstantValue()->AllEqual()); EXPECT_FALSE(sem->ConstantValue()->AnyZero()); EXPECT_FALSE(sem->ConstantValue()->AllZero()); EXPECT_EQ(sem->ConstantValue()->ValueAs(), 99); } TEST_F(ResolverConstEvalTest, Scalar_i32) { auto* expr = Expr(99_i); WrapInFunction(expr); EXPECT_TRUE(r()->Resolve()) << r()->error(); auto* sem = Sem().Get(expr); ASSERT_NE(sem, nullptr); EXPECT_TRUE(sem->Type()->Is()); EXPECT_TYPE(sem->ConstantValue()->Type(), sem->Type()); EXPECT_TRUE(sem->ConstantValue()->AllEqual()); EXPECT_FALSE(sem->ConstantValue()->AnyZero()); EXPECT_FALSE(sem->ConstantValue()->AllZero()); EXPECT_EQ(sem->ConstantValue()->ValueAs(), 99); } TEST_F(ResolverConstEvalTest, Scalar_u32) { auto* expr = Expr(99_u); WrapInFunction(expr); EXPECT_TRUE(r()->Resolve()) << r()->error(); auto* sem = Sem().Get(expr); ASSERT_NE(sem, nullptr); EXPECT_TRUE(sem->Type()->Is()); EXPECT_TYPE(sem->ConstantValue()->Type(), sem->Type()); EXPECT_TRUE(sem->ConstantValue()->AllEqual()); EXPECT_FALSE(sem->ConstantValue()->AnyZero()); EXPECT_FALSE(sem->ConstantValue()->AllZero()); EXPECT_EQ(sem->ConstantValue()->ValueAs(), 99u); } TEST_F(ResolverConstEvalTest, Scalar_f32) { auto* expr = Expr(9.9_f); WrapInFunction(expr); EXPECT_TRUE(r()->Resolve()) << r()->error(); auto* sem = Sem().Get(expr); ASSERT_NE(sem, nullptr); EXPECT_TRUE(sem->Type()->Is()); EXPECT_TYPE(sem->ConstantValue()->Type(), sem->Type()); EXPECT_TRUE(sem->ConstantValue()->AllEqual()); EXPECT_FALSE(sem->ConstantValue()->AnyZero()); EXPECT_FALSE(sem->ConstantValue()->AllZero()); EXPECT_EQ(sem->ConstantValue()->ValueAs().value, 9.9f); } TEST_F(ResolverConstEvalTest, Scalar_f16) { Enable(ast::Extension::kF16); auto* expr = Expr(9.9_h); WrapInFunction(expr); EXPECT_TRUE(r()->Resolve()) << r()->error(); auto* sem = Sem().Get(expr); EXPECT_NE(sem, nullptr); EXPECT_TRUE(sem->Type()->Is()); EXPECT_TYPE(sem->ConstantValue()->Type(), sem->Type()); EXPECT_TRUE(sem->ConstantValue()->AllEqual()); EXPECT_FALSE(sem->ConstantValue()->AnyZero()); EXPECT_FALSE(sem->ConstantValue()->AllZero()); // 9.9 is not exactly representable by f16, and should be quantized to 9.8984375 EXPECT_EQ(sem->ConstantValue()->ValueAs(), 9.8984375f); } TEST_F(ResolverConstEvalTest, Scalar_bool) { auto* expr = Expr(true); WrapInFunction(expr); EXPECT_TRUE(r()->Resolve()) << r()->error(); auto* sem = Sem().Get(expr); ASSERT_NE(sem, nullptr); EXPECT_TRUE(sem->Type()->Is()); EXPECT_TYPE(sem->ConstantValue()->Type(), sem->Type()); EXPECT_TRUE(sem->ConstantValue()->AllEqual()); EXPECT_FALSE(sem->ConstantValue()->AnyZero()); EXPECT_FALSE(sem->ConstantValue()->AllZero()); EXPECT_EQ(sem->ConstantValue()->ValueAs(), true); } namespace ZeroInit { struct Case { builder::ast_type_func_ptr type; }; template Case C() { return Case{builder::DataType::AST}; } using ResolverConstEvalZeroInitTest = ResolverTestWithParam; TEST_P(ResolverConstEvalZeroInitTest, Test) { Enable(ast::Extension::kF16); auto& param = GetParam(); auto* ty = param.type(*this); auto* expr = Construct(ty); auto* a = Const("a", expr); WrapInFunction(a); EXPECT_TRUE(r()->Resolve()) << r()->error(); auto* sem = Sem().Get(expr); ASSERT_NE(sem, nullptr); EXPECT_TRUE(sem->ConstantValue()->AllEqual()); EXPECT_TRUE(sem->ConstantValue()->AnyZero()); EXPECT_TRUE(sem->ConstantValue()->AllZero()); if (sem->Type()->is_scalar()) { EXPECT_EQ(sem->ConstantValue()->Index(0), nullptr); EXPECT_EQ(sem->ConstantValue()->ValueAs(), 0.0f); } else if (auto* vec = sem->Type()->As()) { for (size_t i = 0; i < vec->Width(); ++i) { EXPECT_TRUE(sem->ConstantValue()->Index(i)->AllEqual()); EXPECT_TRUE(sem->ConstantValue()->Index(i)->AnyZero()); EXPECT_TRUE(sem->ConstantValue()->Index(i)->AllZero()); EXPECT_EQ(sem->ConstantValue()->Index(i)->ValueAs(), 0.0f); } } else if (auto* mat = sem->Type()->As()) { for (size_t i = 0; i < mat->columns(); ++i) { EXPECT_TRUE(sem->ConstantValue()->Index(i)->AllEqual()); EXPECT_TRUE(sem->ConstantValue()->Index(i)->AnyZero()); EXPECT_TRUE(sem->ConstantValue()->Index(i)->AllZero()); for (size_t j = 0; j < mat->rows(); ++j) { EXPECT_TRUE(sem->ConstantValue()->Index(i)->Index(j)->AllEqual()); EXPECT_TRUE(sem->ConstantValue()->Index(i)->Index(j)->AnyZero()); EXPECT_TRUE(sem->ConstantValue()->Index(i)->Index(j)->AllZero()); EXPECT_EQ(sem->ConstantValue()->Index(i)->Index(j)->ValueAs(), 0.0f); } } } else if (auto* arr = sem->Type()->As()) { for (size_t i = 0; i < *(arr->ConstantCount()); ++i) { EXPECT_TRUE(sem->ConstantValue()->Index(i)->AllEqual()); EXPECT_TRUE(sem->ConstantValue()->Index(i)->AnyZero()); EXPECT_TRUE(sem->ConstantValue()->Index(i)->AllZero()); EXPECT_EQ(sem->ConstantValue()->Index(i)->ValueAs(), 0.0f); } } } INSTANTIATE_TEST_SUITE_P(ZeroInit, ResolverConstEvalZeroInitTest, testing::ValuesIn({ C(), C(), C(), C(), C(), C>(), C>(), C>(), C>(), C>(), C>(), C>(), C>(), C>(), C>(), C>(), })); } // namespace ZeroInit TEST_F(ResolverConstEvalTest, Vec3_ZeroInit_i32) { auto* expr = vec3(); WrapInFunction(expr); EXPECT_TRUE(r()->Resolve()) << r()->error(); auto* sem = Sem().Get(expr); ASSERT_NE(sem, nullptr); auto* vec = sem->Type()->As(); ASSERT_NE(vec, nullptr); EXPECT_TRUE(vec->type()->Is()); EXPECT_EQ(vec->Width(), 3u); EXPECT_TYPE(sem->ConstantValue()->Type(), sem->Type()); EXPECT_TRUE(sem->ConstantValue()->AllEqual()); EXPECT_TRUE(sem->ConstantValue()->AnyZero()); EXPECT_TRUE(sem->ConstantValue()->AllZero()); EXPECT_TRUE(sem->ConstantValue()->Index(0)->AllEqual()); EXPECT_TRUE(sem->ConstantValue()->Index(0)->AnyZero()); EXPECT_TRUE(sem->ConstantValue()->Index(0)->AllZero()); EXPECT_EQ(sem->ConstantValue()->Index(0)->ValueAs(), 0); EXPECT_TRUE(sem->ConstantValue()->Index(1)->AllEqual()); EXPECT_TRUE(sem->ConstantValue()->Index(1)->AnyZero()); EXPECT_TRUE(sem->ConstantValue()->Index(1)->AllZero()); EXPECT_EQ(sem->ConstantValue()->Index(1)->ValueAs(), 0); EXPECT_TRUE(sem->ConstantValue()->Index(2)->AllEqual()); EXPECT_TRUE(sem->ConstantValue()->Index(2)->AnyZero()); EXPECT_TRUE(sem->ConstantValue()->Index(2)->AllZero()); EXPECT_EQ(sem->ConstantValue()->Index(2)->ValueAs(), 0); } TEST_F(ResolverConstEvalTest, Vec3_ZeroInit_u32) { auto* expr = vec3(); WrapInFunction(expr); EXPECT_TRUE(r()->Resolve()) << r()->error(); auto* sem = Sem().Get(expr); ASSERT_NE(sem, nullptr); auto* vec = sem->Type()->As(); ASSERT_NE(vec, nullptr); EXPECT_TRUE(vec->type()->Is()); EXPECT_EQ(vec->Width(), 3u); EXPECT_TYPE(sem->ConstantValue()->Type(), sem->Type()); EXPECT_TRUE(sem->ConstantValue()->AllEqual()); EXPECT_TRUE(sem->ConstantValue()->AnyZero()); EXPECT_TRUE(sem->ConstantValue()->AllZero()); EXPECT_TRUE(sem->ConstantValue()->Index(0)->AllEqual()); EXPECT_TRUE(sem->ConstantValue()->Index(0)->AnyZero()); EXPECT_TRUE(sem->ConstantValue()->Index(0)->AllZero()); EXPECT_EQ(sem->ConstantValue()->Index(0)->ValueAs(), 0u); EXPECT_TRUE(sem->ConstantValue()->Index(1)->AllEqual()); EXPECT_TRUE(sem->ConstantValue()->Index(1)->AnyZero()); EXPECT_TRUE(sem->ConstantValue()->Index(1)->AllZero()); EXPECT_EQ(sem->ConstantValue()->Index(1)->ValueAs(), 0u); EXPECT_TRUE(sem->ConstantValue()->Index(2)->AllEqual()); EXPECT_TRUE(sem->ConstantValue()->Index(2)->AnyZero()); EXPECT_TRUE(sem->ConstantValue()->Index(2)->AllZero()); EXPECT_EQ(sem->ConstantValue()->Index(2)->ValueAs(), 0u); } TEST_F(ResolverConstEvalTest, Vec3_ZeroInit_f32) { auto* expr = vec3(); WrapInFunction(expr); EXPECT_TRUE(r()->Resolve()) << r()->error(); auto* sem = Sem().Get(expr); ASSERT_NE(sem, nullptr); auto* vec = sem->Type()->As(); ASSERT_NE(vec, nullptr); EXPECT_TRUE(vec->type()->Is()); EXPECT_EQ(vec->Width(), 3u); EXPECT_TYPE(sem->ConstantValue()->Type(), sem->Type()); EXPECT_TRUE(sem->ConstantValue()->AllEqual()); EXPECT_TRUE(sem->ConstantValue()->AnyZero()); EXPECT_TRUE(sem->ConstantValue()->AllZero()); EXPECT_TRUE(sem->ConstantValue()->Index(0)->AllEqual()); EXPECT_TRUE(sem->ConstantValue()->Index(0)->AnyZero()); EXPECT_TRUE(sem->ConstantValue()->Index(0)->AllZero()); EXPECT_EQ(sem->ConstantValue()->Index(0)->ValueAs(), 0._a); EXPECT_TRUE(sem->ConstantValue()->Index(1)->AllEqual()); EXPECT_TRUE(sem->ConstantValue()->Index(1)->AnyZero()); EXPECT_TRUE(sem->ConstantValue()->Index(1)->AllZero()); EXPECT_EQ(sem->ConstantValue()->Index(1)->ValueAs(), 0._a); EXPECT_TRUE(sem->ConstantValue()->Index(2)->AllEqual()); EXPECT_TRUE(sem->ConstantValue()->Index(2)->AnyZero()); EXPECT_TRUE(sem->ConstantValue()->Index(2)->AllZero()); EXPECT_EQ(sem->ConstantValue()->Index(2)->ValueAs(), 0._a); } TEST_F(ResolverConstEvalTest, Vec3_ZeroInit_f16) { Enable(ast::Extension::kF16); auto* expr = vec3(); WrapInFunction(expr); EXPECT_TRUE(r()->Resolve()) << r()->error(); auto* sem = Sem().Get(expr); EXPECT_NE(sem, nullptr); auto* vec = sem->Type()->As(); ASSERT_NE(vec, nullptr); EXPECT_TRUE(vec->type()->Is()); EXPECT_EQ(vec->Width(), 3u); EXPECT_TYPE(sem->ConstantValue()->Type(), sem->Type()); EXPECT_TRUE(sem->ConstantValue()->AllEqual()); EXPECT_TRUE(sem->ConstantValue()->AnyZero()); EXPECT_TRUE(sem->ConstantValue()->AllZero()); EXPECT_TRUE(sem->ConstantValue()->Index(0)->AllEqual()); EXPECT_TRUE(sem->ConstantValue()->Index(0)->AnyZero()); EXPECT_TRUE(sem->ConstantValue()->Index(0)->AllZero()); EXPECT_EQ(sem->ConstantValue()->Index(0)->ValueAs(), 0._a); EXPECT_TRUE(sem->ConstantValue()->Index(1)->AllEqual()); EXPECT_TRUE(sem->ConstantValue()->Index(1)->AnyZero()); EXPECT_TRUE(sem->ConstantValue()->Index(1)->AllZero()); EXPECT_EQ(sem->ConstantValue()->Index(1)->ValueAs(), 0._a); EXPECT_TRUE(sem->ConstantValue()->Index(2)->AllEqual()); EXPECT_TRUE(sem->ConstantValue()->Index(2)->AnyZero()); EXPECT_TRUE(sem->ConstantValue()->Index(2)->AllZero()); EXPECT_EQ(sem->ConstantValue()->Index(2)->ValueAs(), 0._a); } TEST_F(ResolverConstEvalTest, Vec3_ZeroInit_bool) { auto* expr = vec3(); WrapInFunction(expr); EXPECT_TRUE(r()->Resolve()) << r()->error(); auto* sem = Sem().Get(expr); ASSERT_NE(sem, nullptr); auto* vec = sem->Type()->As(); ASSERT_NE(vec, nullptr); EXPECT_TRUE(vec->type()->Is()); EXPECT_EQ(vec->Width(), 3u); EXPECT_TYPE(sem->ConstantValue()->Type(), sem->Type()); EXPECT_TRUE(sem->ConstantValue()->AllEqual()); EXPECT_TRUE(sem->ConstantValue()->AnyZero()); EXPECT_TRUE(sem->ConstantValue()->AllZero()); EXPECT_TRUE(sem->ConstantValue()->Index(0)->AllEqual()); EXPECT_TRUE(sem->ConstantValue()->Index(0)->AnyZero()); EXPECT_TRUE(sem->ConstantValue()->Index(0)->AllZero()); EXPECT_EQ(sem->ConstantValue()->Index(0)->ValueAs(), false); EXPECT_TRUE(sem->ConstantValue()->Index(1)->AllEqual()); EXPECT_TRUE(sem->ConstantValue()->Index(1)->AnyZero()); EXPECT_TRUE(sem->ConstantValue()->Index(1)->AllZero()); EXPECT_EQ(sem->ConstantValue()->Index(1)->ValueAs(), false); EXPECT_TRUE(sem->ConstantValue()->Index(2)->AllEqual()); EXPECT_TRUE(sem->ConstantValue()->Index(2)->AnyZero()); EXPECT_TRUE(sem->ConstantValue()->Index(2)->AllZero()); EXPECT_EQ(sem->ConstantValue()->Index(2)->ValueAs(), false); } TEST_F(ResolverConstEvalTest, Vec3_Splat_i32) { auto* expr = vec3(99_i); WrapInFunction(expr); EXPECT_TRUE(r()->Resolve()) << r()->error(); auto* sem = Sem().Get(expr); ASSERT_NE(sem, nullptr); auto* vec = sem->Type()->As(); ASSERT_NE(vec, nullptr); EXPECT_TRUE(vec->type()->Is()); EXPECT_EQ(vec->Width(), 3u); EXPECT_TYPE(sem->ConstantValue()->Type(), sem->Type()); EXPECT_TRUE(sem->ConstantValue()->AllEqual()); EXPECT_FALSE(sem->ConstantValue()->AnyZero()); EXPECT_FALSE(sem->ConstantValue()->AllZero()); EXPECT_TRUE(sem->ConstantValue()->Index(0)->AllEqual()); EXPECT_FALSE(sem->ConstantValue()->Index(0)->AnyZero()); EXPECT_FALSE(sem->ConstantValue()->Index(0)->AllZero()); EXPECT_EQ(sem->ConstantValue()->Index(0)->ValueAs(), 99); EXPECT_TRUE(sem->ConstantValue()->Index(1)->AllEqual()); EXPECT_FALSE(sem->ConstantValue()->Index(1)->AnyZero()); EXPECT_FALSE(sem->ConstantValue()->Index(1)->AllZero()); EXPECT_EQ(sem->ConstantValue()->Index(1)->ValueAs(), 99); EXPECT_TRUE(sem->ConstantValue()->Index(2)->AllEqual()); EXPECT_FALSE(sem->ConstantValue()->Index(2)->AnyZero()); EXPECT_FALSE(sem->ConstantValue()->Index(2)->AllZero()); EXPECT_EQ(sem->ConstantValue()->Index(2)->ValueAs(), 99); } TEST_F(ResolverConstEvalTest, Vec3_Splat_u32) { auto* expr = vec3(99_u); WrapInFunction(expr); EXPECT_TRUE(r()->Resolve()) << r()->error(); auto* sem = Sem().Get(expr); ASSERT_NE(sem, nullptr); auto* vec = sem->Type()->As(); ASSERT_NE(vec, nullptr); EXPECT_TRUE(vec->type()->Is()); EXPECT_EQ(vec->Width(), 3u); EXPECT_TYPE(sem->ConstantValue()->Type(), sem->Type()); EXPECT_TRUE(sem->ConstantValue()->AllEqual()); EXPECT_FALSE(sem->ConstantValue()->AnyZero()); EXPECT_FALSE(sem->ConstantValue()->AllZero()); EXPECT_TRUE(sem->ConstantValue()->Index(0)->AllEqual()); EXPECT_FALSE(sem->ConstantValue()->Index(0)->AnyZero()); EXPECT_FALSE(sem->ConstantValue()->Index(0)->AllZero()); EXPECT_EQ(sem->ConstantValue()->Index(0)->ValueAs(), 99u); EXPECT_TRUE(sem->ConstantValue()->Index(1)->AllEqual()); EXPECT_FALSE(sem->ConstantValue()->Index(1)->AnyZero()); EXPECT_FALSE(sem->ConstantValue()->Index(1)->AllZero()); EXPECT_EQ(sem->ConstantValue()->Index(1)->ValueAs(), 99u); EXPECT_TRUE(sem->ConstantValue()->Index(2)->AllEqual()); EXPECT_FALSE(sem->ConstantValue()->Index(2)->AnyZero()); EXPECT_FALSE(sem->ConstantValue()->Index(2)->AllZero()); EXPECT_EQ(sem->ConstantValue()->Index(2)->ValueAs(), 99u); } TEST_F(ResolverConstEvalTest, Vec3_Splat_f32) { auto* expr = vec3(9.9_f); WrapInFunction(expr); EXPECT_TRUE(r()->Resolve()) << r()->error(); auto* sem = Sem().Get(expr); ASSERT_NE(sem, nullptr); auto* vec = sem->Type()->As(); ASSERT_NE(vec, nullptr); EXPECT_TRUE(vec->type()->Is()); EXPECT_EQ(vec->Width(), 3u); EXPECT_TYPE(sem->ConstantValue()->Type(), sem->Type()); EXPECT_TRUE(sem->ConstantValue()->AllEqual()); EXPECT_FALSE(sem->ConstantValue()->AnyZero()); EXPECT_FALSE(sem->ConstantValue()->AllZero()); EXPECT_TRUE(sem->ConstantValue()->Index(0)->AllEqual()); EXPECT_FALSE(sem->ConstantValue()->Index(0)->AnyZero()); EXPECT_FALSE(sem->ConstantValue()->Index(0)->AllZero()); EXPECT_EQ(sem->ConstantValue()->Index(0)->ValueAs(), 9.9f); EXPECT_TRUE(sem->ConstantValue()->Index(1)->AllEqual()); EXPECT_FALSE(sem->ConstantValue()->Index(1)->AnyZero()); EXPECT_FALSE(sem->ConstantValue()->Index(1)->AllZero()); EXPECT_EQ(sem->ConstantValue()->Index(1)->ValueAs(), 9.9f); EXPECT_TRUE(sem->ConstantValue()->Index(2)->AllEqual()); EXPECT_FALSE(sem->ConstantValue()->Index(2)->AnyZero()); EXPECT_FALSE(sem->ConstantValue()->Index(2)->AllZero()); EXPECT_EQ(sem->ConstantValue()->Index(2)->ValueAs(), 9.9f); } TEST_F(ResolverConstEvalTest, Vec3_Splat_f16) { Enable(ast::Extension::kF16); auto* expr = vec3(9.9_h); WrapInFunction(expr); EXPECT_TRUE(r()->Resolve()) << r()->error(); auto* sem = Sem().Get(expr); EXPECT_NE(sem, nullptr); auto* vec = sem->Type()->As(); ASSERT_NE(vec, nullptr); EXPECT_TRUE(vec->type()->Is()); EXPECT_EQ(vec->Width(), 3u); EXPECT_TYPE(sem->ConstantValue()->Type(), sem->Type()); EXPECT_TRUE(sem->ConstantValue()->AllEqual()); EXPECT_FALSE(sem->ConstantValue()->AnyZero()); EXPECT_FALSE(sem->ConstantValue()->AllZero()); // 9.9 is not exactly representable by f16, and should be quantized to 9.8984375 EXPECT_TRUE(sem->ConstantValue()->Index(0)->AllEqual()); EXPECT_FALSE(sem->ConstantValue()->Index(0)->AnyZero()); EXPECT_FALSE(sem->ConstantValue()->Index(0)->AllZero()); EXPECT_EQ(sem->ConstantValue()->Index(0)->ValueAs(), 9.8984375f); EXPECT_TRUE(sem->ConstantValue()->Index(1)->AllEqual()); EXPECT_FALSE(sem->ConstantValue()->Index(1)->AnyZero()); EXPECT_FALSE(sem->ConstantValue()->Index(1)->AllZero()); EXPECT_EQ(sem->ConstantValue()->Index(1)->ValueAs(), 9.8984375f); EXPECT_TRUE(sem->ConstantValue()->Index(2)->AllEqual()); EXPECT_FALSE(sem->ConstantValue()->Index(2)->AnyZero()); EXPECT_FALSE(sem->ConstantValue()->Index(2)->AllZero()); EXPECT_EQ(sem->ConstantValue()->Index(2)->ValueAs(), 9.8984375f); } TEST_F(ResolverConstEvalTest, Vec3_Splat_bool) { auto* expr = vec3(true); WrapInFunction(expr); EXPECT_TRUE(r()->Resolve()) << r()->error(); auto* sem = Sem().Get(expr); ASSERT_NE(sem, nullptr); auto* vec = sem->Type()->As(); ASSERT_NE(vec, nullptr); EXPECT_TRUE(vec->type()->Is()); EXPECT_EQ(vec->Width(), 3u); EXPECT_TYPE(sem->ConstantValue()->Type(), sem->Type()); EXPECT_TRUE(sem->ConstantValue()->AllEqual()); EXPECT_FALSE(sem->ConstantValue()->AnyZero()); EXPECT_FALSE(sem->ConstantValue()->AllZero()); EXPECT_TRUE(sem->ConstantValue()->Index(0)->AllEqual()); EXPECT_FALSE(sem->ConstantValue()->Index(0)->AnyZero()); EXPECT_FALSE(sem->ConstantValue()->Index(0)->AllZero()); EXPECT_EQ(sem->ConstantValue()->Index(0)->ValueAs(), true); EXPECT_TRUE(sem->ConstantValue()->Index(1)->AllEqual()); EXPECT_FALSE(sem->ConstantValue()->Index(1)->AnyZero()); EXPECT_FALSE(sem->ConstantValue()->Index(1)->AllZero()); EXPECT_EQ(sem->ConstantValue()->Index(1)->ValueAs(), true); EXPECT_TRUE(sem->ConstantValue()->Index(2)->AllEqual()); EXPECT_FALSE(sem->ConstantValue()->Index(2)->AnyZero()); EXPECT_FALSE(sem->ConstantValue()->Index(2)->AllZero()); EXPECT_EQ(sem->ConstantValue()->Index(2)->ValueAs(), true); } TEST_F(ResolverConstEvalTest, Vec3_FullConstruct_AInt) { auto* expr = vec3(1_a, 2_a, 3_a); auto* a = Const("a", expr); WrapInFunction(a); EXPECT_TRUE(r()->Resolve()) << r()->error(); auto* sem = Sem().Get(expr); ASSERT_NE(sem, nullptr); auto* vec = sem->Type()->As(); ASSERT_NE(vec, nullptr); EXPECT_TRUE(vec->type()->Is()); EXPECT_EQ(vec->Width(), 3u); EXPECT_TYPE(sem->ConstantValue()->Type(), sem->Type()); EXPECT_FALSE(sem->ConstantValue()->AllEqual()); EXPECT_FALSE(sem->ConstantValue()->AnyZero()); EXPECT_FALSE(sem->ConstantValue()->AllZero()); EXPECT_TRUE(sem->ConstantValue()->Index(0)->AllEqual()); EXPECT_FALSE(sem->ConstantValue()->Index(0)->AnyZero()); EXPECT_FALSE(sem->ConstantValue()->Index(0)->AllZero()); EXPECT_EQ(sem->ConstantValue()->Index(0)->ValueAs(), 1); EXPECT_TRUE(sem->ConstantValue()->Index(1)->AllEqual()); EXPECT_FALSE(sem->ConstantValue()->Index(1)->AnyZero()); EXPECT_FALSE(sem->ConstantValue()->Index(1)->AllZero()); EXPECT_EQ(sem->ConstantValue()->Index(1)->ValueAs(), 2); EXPECT_TRUE(sem->ConstantValue()->Index(2)->AllEqual()); EXPECT_FALSE(sem->ConstantValue()->Index(2)->AnyZero()); EXPECT_FALSE(sem->ConstantValue()->Index(2)->AllZero()); EXPECT_EQ(sem->ConstantValue()->Index(2)->ValueAs(), 3); } TEST_F(ResolverConstEvalTest, Vec3_FullConstruct_AFloat) { auto* expr = vec3(1.0_a, 2.0_a, 3.0_a); auto* a = Const("a", expr); WrapInFunction(a); EXPECT_TRUE(r()->Resolve()) << r()->error(); auto* sem = Sem().Get(expr); ASSERT_NE(sem, nullptr); auto* vec = sem->Type()->As(); ASSERT_NE(vec, nullptr); EXPECT_TRUE(vec->type()->Is()); EXPECT_EQ(vec->Width(), 3u); EXPECT_TYPE(sem->ConstantValue()->Type(), sem->Type()); EXPECT_FALSE(sem->ConstantValue()->AllEqual()); EXPECT_FALSE(sem->ConstantValue()->AnyZero()); EXPECT_FALSE(sem->ConstantValue()->AllZero()); EXPECT_TRUE(sem->ConstantValue()->Index(0)->AllEqual()); EXPECT_FALSE(sem->ConstantValue()->Index(0)->AnyZero()); EXPECT_FALSE(sem->ConstantValue()->Index(0)->AllZero()); EXPECT_EQ(sem->ConstantValue()->Index(0)->ValueAs(), 1.0f); EXPECT_TRUE(sem->ConstantValue()->Index(1)->AllEqual()); EXPECT_FALSE(sem->ConstantValue()->Index(1)->AnyZero()); EXPECT_FALSE(sem->ConstantValue()->Index(1)->AllZero()); EXPECT_EQ(sem->ConstantValue()->Index(1)->ValueAs(), 2.0f); EXPECT_TRUE(sem->ConstantValue()->Index(2)->AllEqual()); EXPECT_FALSE(sem->ConstantValue()->Index(2)->AnyZero()); EXPECT_FALSE(sem->ConstantValue()->Index(2)->AllZero()); EXPECT_EQ(sem->ConstantValue()->Index(2)->ValueAs(), 3.0f); } TEST_F(ResolverConstEvalTest, Vec3_FullConstruct_i32) { auto* expr = vec3(1_i, 2_i, 3_i); WrapInFunction(expr); EXPECT_TRUE(r()->Resolve()) << r()->error(); auto* sem = Sem().Get(expr); ASSERT_NE(sem, nullptr); auto* vec = sem->Type()->As(); ASSERT_NE(vec, nullptr); EXPECT_TRUE(vec->type()->Is()); EXPECT_EQ(vec->Width(), 3u); EXPECT_TYPE(sem->ConstantValue()->Type(), sem->Type()); EXPECT_FALSE(sem->ConstantValue()->AllEqual()); EXPECT_FALSE(sem->ConstantValue()->AnyZero()); EXPECT_FALSE(sem->ConstantValue()->AllZero()); EXPECT_TRUE(sem->ConstantValue()->Index(0)->AllEqual()); EXPECT_FALSE(sem->ConstantValue()->Index(0)->AnyZero()); EXPECT_FALSE(sem->ConstantValue()->Index(0)->AllZero()); EXPECT_EQ(sem->ConstantValue()->Index(0)->ValueAs(), 1); EXPECT_TRUE(sem->ConstantValue()->Index(1)->AllEqual()); EXPECT_FALSE(sem->ConstantValue()->Index(1)->AnyZero()); EXPECT_FALSE(sem->ConstantValue()->Index(1)->AllZero()); EXPECT_EQ(sem->ConstantValue()->Index(1)->ValueAs(), 2); EXPECT_TRUE(sem->ConstantValue()->Index(2)->AllEqual()); EXPECT_FALSE(sem->ConstantValue()->Index(2)->AnyZero()); EXPECT_FALSE(sem->ConstantValue()->Index(2)->AllZero()); EXPECT_EQ(sem->ConstantValue()->Index(2)->ValueAs(), 3); } TEST_F(ResolverConstEvalTest, Vec3_FullConstruct_u32) { auto* expr = vec3(1_u, 2_u, 3_u); WrapInFunction(expr); EXPECT_TRUE(r()->Resolve()) << r()->error(); auto* sem = Sem().Get(expr); ASSERT_NE(sem, nullptr); auto* vec = sem->Type()->As(); ASSERT_NE(vec, nullptr); EXPECT_TRUE(vec->type()->Is()); EXPECT_EQ(vec->Width(), 3u); EXPECT_TYPE(sem->ConstantValue()->Type(), sem->Type()); EXPECT_FALSE(sem->ConstantValue()->AllEqual()); EXPECT_FALSE(sem->ConstantValue()->AnyZero()); EXPECT_FALSE(sem->ConstantValue()->AllZero()); EXPECT_TRUE(sem->ConstantValue()->Index(0)->AllEqual()); EXPECT_FALSE(sem->ConstantValue()->Index(0)->AnyZero()); EXPECT_FALSE(sem->ConstantValue()->Index(0)->AllZero()); EXPECT_EQ(sem->ConstantValue()->Index(0)->ValueAs(), 1); EXPECT_TRUE(sem->ConstantValue()->Index(1)->AllEqual()); EXPECT_FALSE(sem->ConstantValue()->Index(1)->AnyZero()); EXPECT_FALSE(sem->ConstantValue()->Index(1)->AllZero()); EXPECT_EQ(sem->ConstantValue()->Index(1)->ValueAs(), 2); EXPECT_TRUE(sem->ConstantValue()->Index(2)->AllEqual()); EXPECT_FALSE(sem->ConstantValue()->Index(2)->AnyZero()); EXPECT_FALSE(sem->ConstantValue()->Index(2)->AllZero()); EXPECT_EQ(sem->ConstantValue()->Index(2)->ValueAs(), 3); } TEST_F(ResolverConstEvalTest, Vec3_FullConstruct_f32) { auto* expr = vec3(1_f, 2_f, 3_f); WrapInFunction(expr); EXPECT_TRUE(r()->Resolve()) << r()->error(); auto* sem = Sem().Get(expr); ASSERT_NE(sem, nullptr); auto* vec = sem->Type()->As(); ASSERT_NE(vec, nullptr); EXPECT_TRUE(vec->type()->Is()); EXPECT_EQ(vec->Width(), 3u); EXPECT_TYPE(sem->ConstantValue()->Type(), sem->Type()); EXPECT_FALSE(sem->ConstantValue()->AllEqual()); EXPECT_FALSE(sem->ConstantValue()->AnyZero()); EXPECT_FALSE(sem->ConstantValue()->AllZero()); EXPECT_TRUE(sem->ConstantValue()->Index(0)->AllEqual()); EXPECT_FALSE(sem->ConstantValue()->Index(0)->AnyZero()); EXPECT_FALSE(sem->ConstantValue()->Index(0)->AllZero()); EXPECT_EQ(sem->ConstantValue()->Index(0)->ValueAs(), 1.f); EXPECT_TRUE(sem->ConstantValue()->Index(1)->AllEqual()); EXPECT_FALSE(sem->ConstantValue()->Index(1)->AnyZero()); EXPECT_FALSE(sem->ConstantValue()->Index(1)->AllZero()); EXPECT_EQ(sem->ConstantValue()->Index(1)->ValueAs(), 2.f); EXPECT_TRUE(sem->ConstantValue()->Index(2)->AllEqual()); EXPECT_FALSE(sem->ConstantValue()->Index(2)->AnyZero()); EXPECT_FALSE(sem->ConstantValue()->Index(2)->AllZero()); EXPECT_EQ(sem->ConstantValue()->Index(2)->ValueAs(), 3.f); } TEST_F(ResolverConstEvalTest, Vec3_FullConstruct_f16) { Enable(ast::Extension::kF16); auto* expr = vec3(1_h, 2_h, 3_h); WrapInFunction(expr); EXPECT_TRUE(r()->Resolve()) << r()->error(); auto* sem = Sem().Get(expr); EXPECT_NE(sem, nullptr); auto* vec = sem->Type()->As(); ASSERT_NE(vec, nullptr); EXPECT_TRUE(vec->type()->Is()); EXPECT_EQ(vec->Width(), 3u); EXPECT_TYPE(sem->ConstantValue()->Type(), sem->Type()); EXPECT_FALSE(sem->ConstantValue()->AllEqual()); EXPECT_FALSE(sem->ConstantValue()->AnyZero()); EXPECT_FALSE(sem->ConstantValue()->AllZero()); EXPECT_TRUE(sem->ConstantValue()->Index(0)->AllEqual()); EXPECT_FALSE(sem->ConstantValue()->Index(0)->AnyZero()); EXPECT_FALSE(sem->ConstantValue()->Index(0)->AllZero()); EXPECT_EQ(sem->ConstantValue()->Index(0)->ValueAs(), 1.f); EXPECT_TRUE(sem->ConstantValue()->Index(1)->AllEqual()); EXPECT_FALSE(sem->ConstantValue()->Index(1)->AnyZero()); EXPECT_FALSE(sem->ConstantValue()->Index(1)->AllZero()); EXPECT_EQ(sem->ConstantValue()->Index(1)->ValueAs(), 2.f); EXPECT_TRUE(sem->ConstantValue()->Index(2)->AllEqual()); EXPECT_FALSE(sem->ConstantValue()->Index(2)->AnyZero()); EXPECT_FALSE(sem->ConstantValue()->Index(2)->AllZero()); EXPECT_EQ(sem->ConstantValue()->Index(2)->ValueAs(), 3.f); } TEST_F(ResolverConstEvalTest, Vec3_FullConstruct_bool) { auto* expr = vec3(true, false, true); WrapInFunction(expr); EXPECT_TRUE(r()->Resolve()) << r()->error(); auto* sem = Sem().Get(expr); ASSERT_NE(sem, nullptr); auto* vec = sem->Type()->As(); ASSERT_NE(vec, nullptr); EXPECT_TRUE(vec->type()->Is()); EXPECT_EQ(vec->Width(), 3u); EXPECT_TYPE(sem->ConstantValue()->Type(), sem->Type()); EXPECT_FALSE(sem->ConstantValue()->AllEqual()); EXPECT_TRUE(sem->ConstantValue()->AnyZero()); EXPECT_FALSE(sem->ConstantValue()->AllZero()); EXPECT_TRUE(sem->ConstantValue()->Index(0)->AllEqual()); EXPECT_FALSE(sem->ConstantValue()->Index(0)->AnyZero()); EXPECT_FALSE(sem->ConstantValue()->Index(0)->AllZero()); EXPECT_EQ(sem->ConstantValue()->Index(0)->ValueAs(), true); EXPECT_TRUE(sem->ConstantValue()->Index(1)->AllEqual()); EXPECT_TRUE(sem->ConstantValue()->Index(1)->AnyZero()); EXPECT_TRUE(sem->ConstantValue()->Index(1)->AllZero()); EXPECT_EQ(sem->ConstantValue()->Index(1)->ValueAs(), false); EXPECT_TRUE(sem->ConstantValue()->Index(2)->AllEqual()); EXPECT_FALSE(sem->ConstantValue()->Index(2)->AnyZero()); EXPECT_FALSE(sem->ConstantValue()->Index(2)->AllZero()); EXPECT_EQ(sem->ConstantValue()->Index(2)->ValueAs(), true); } TEST_F(ResolverConstEvalTest, Vec3_MixConstruct_i32) { auto* expr = vec3(1_i, vec2(2_i, 3_i)); WrapInFunction(expr); EXPECT_TRUE(r()->Resolve()) << r()->error(); auto* sem = Sem().Get(expr); ASSERT_NE(sem, nullptr); auto* vec = sem->Type()->As(); ASSERT_NE(vec, nullptr); EXPECT_TRUE(vec->type()->Is()); EXPECT_EQ(vec->Width(), 3u); EXPECT_TYPE(sem->ConstantValue()->Type(), sem->Type()); EXPECT_FALSE(sem->ConstantValue()->AllEqual()); EXPECT_FALSE(sem->ConstantValue()->AnyZero()); EXPECT_FALSE(sem->ConstantValue()->AllZero()); EXPECT_TRUE(sem->ConstantValue()->Index(0)->AllEqual()); EXPECT_FALSE(sem->ConstantValue()->Index(0)->AnyZero()); EXPECT_FALSE(sem->ConstantValue()->Index(0)->AllZero()); EXPECT_EQ(sem->ConstantValue()->Index(0)->ValueAs(), 1); EXPECT_TRUE(sem->ConstantValue()->Index(1)->AllEqual()); EXPECT_FALSE(sem->ConstantValue()->Index(1)->AnyZero()); EXPECT_FALSE(sem->ConstantValue()->Index(1)->AllZero()); EXPECT_EQ(sem->ConstantValue()->Index(1)->ValueAs(), 2); EXPECT_TRUE(sem->ConstantValue()->Index(2)->AllEqual()); EXPECT_FALSE(sem->ConstantValue()->Index(2)->AnyZero()); EXPECT_FALSE(sem->ConstantValue()->Index(2)->AllZero()); EXPECT_EQ(sem->ConstantValue()->Index(2)->ValueAs(), 3); } TEST_F(ResolverConstEvalTest, Vec3_MixConstruct_u32) { auto* expr = vec3(vec2(1_u, 2_u), 3_u); WrapInFunction(expr); EXPECT_TRUE(r()->Resolve()) << r()->error(); auto* sem = Sem().Get(expr); ASSERT_NE(sem, nullptr); auto* vec = sem->Type()->As(); ASSERT_NE(vec, nullptr); EXPECT_TRUE(vec->type()->Is()); EXPECT_EQ(vec->Width(), 3u); EXPECT_TYPE(sem->ConstantValue()->Type(), sem->Type()); EXPECT_FALSE(sem->ConstantValue()->AllEqual()); EXPECT_FALSE(sem->ConstantValue()->AnyZero()); EXPECT_FALSE(sem->ConstantValue()->AllZero()); EXPECT_TRUE(sem->ConstantValue()->Index(0)->AllEqual()); EXPECT_FALSE(sem->ConstantValue()->Index(0)->AnyZero()); EXPECT_FALSE(sem->ConstantValue()->Index(0)->AllZero()); EXPECT_EQ(sem->ConstantValue()->Index(0)->ValueAs(), 1); EXPECT_TRUE(sem->ConstantValue()->Index(1)->AllEqual()); EXPECT_FALSE(sem->ConstantValue()->Index(1)->AnyZero()); EXPECT_FALSE(sem->ConstantValue()->Index(1)->AllZero()); EXPECT_EQ(sem->ConstantValue()->Index(1)->ValueAs(), 2); EXPECT_TRUE(sem->ConstantValue()->Index(2)->AllEqual()); EXPECT_FALSE(sem->ConstantValue()->Index(2)->AnyZero()); EXPECT_FALSE(sem->ConstantValue()->Index(2)->AllZero()); EXPECT_EQ(sem->ConstantValue()->Index(2)->ValueAs(), 3); } TEST_F(ResolverConstEvalTest, Vec3_MixConstruct_f32) { auto* expr = vec3(1_f, vec2(2_f, 3_f)); WrapInFunction(expr); EXPECT_TRUE(r()->Resolve()) << r()->error(); auto* sem = Sem().Get(expr); ASSERT_NE(sem, nullptr); auto* vec = sem->Type()->As(); ASSERT_NE(vec, nullptr); EXPECT_TRUE(vec->type()->Is()); EXPECT_EQ(vec->Width(), 3u); EXPECT_TYPE(sem->ConstantValue()->Type(), sem->Type()); EXPECT_FALSE(sem->ConstantValue()->AllEqual()); EXPECT_FALSE(sem->ConstantValue()->AnyZero()); EXPECT_FALSE(sem->ConstantValue()->AllZero()); EXPECT_TRUE(sem->ConstantValue()->Index(0)->AllEqual()); EXPECT_FALSE(sem->ConstantValue()->Index(0)->AnyZero()); EXPECT_FALSE(sem->ConstantValue()->Index(0)->AllZero()); EXPECT_EQ(sem->ConstantValue()->Index(0)->ValueAs(), 1.f); EXPECT_TRUE(sem->ConstantValue()->Index(1)->AllEqual()); EXPECT_FALSE(sem->ConstantValue()->Index(1)->AnyZero()); EXPECT_FALSE(sem->ConstantValue()->Index(1)->AllZero()); EXPECT_EQ(sem->ConstantValue()->Index(1)->ValueAs(), 2.f); EXPECT_TRUE(sem->ConstantValue()->Index(2)->AllEqual()); EXPECT_FALSE(sem->ConstantValue()->Index(2)->AnyZero()); EXPECT_FALSE(sem->ConstantValue()->Index(2)->AllZero()); EXPECT_EQ(sem->ConstantValue()->Index(2)->ValueAs(), 3.f); } TEST_F(ResolverConstEvalTest, Vec3_MixConstruct_f32_all_10) { auto* expr = vec3(10_f, vec2(10_f, 10_f)); WrapInFunction(expr); EXPECT_TRUE(r()->Resolve()) << r()->error(); auto* sem = Sem().Get(expr); EXPECT_NE(sem, nullptr); auto* vec = sem->Type()->As(); ASSERT_NE(vec, nullptr); EXPECT_TRUE(vec->type()->Is()); EXPECT_EQ(vec->Width(), 3u); EXPECT_TYPE(sem->ConstantValue()->Type(), sem->Type()); EXPECT_TRUE(sem->ConstantValue()->AllEqual()); EXPECT_FALSE(sem->ConstantValue()->AnyZero()); EXPECT_FALSE(sem->ConstantValue()->AllZero()); EXPECT_TRUE(sem->ConstantValue()->Index(0)->AllEqual()); EXPECT_FALSE(sem->ConstantValue()->Index(0)->AnyZero()); EXPECT_FALSE(sem->ConstantValue()->Index(0)->AllZero()); EXPECT_EQ(sem->ConstantValue()->Index(0)->ValueAs(), 10_f); EXPECT_TRUE(sem->ConstantValue()->Index(1)->AllEqual()); EXPECT_FALSE(sem->ConstantValue()->Index(1)->AnyZero()); EXPECT_FALSE(sem->ConstantValue()->Index(1)->AllZero()); EXPECT_EQ(sem->ConstantValue()->Index(1)->ValueAs(), 10_f); EXPECT_TRUE(sem->ConstantValue()->Index(2)->AllEqual()); EXPECT_FALSE(sem->ConstantValue()->Index(2)->AnyZero()); EXPECT_FALSE(sem->ConstantValue()->Index(2)->AllZero()); EXPECT_EQ(sem->ConstantValue()->Index(2)->ValueAs(), 10_f); } TEST_F(ResolverConstEvalTest, Vec3_MixConstruct_f32_all_positive_0) { auto* expr = vec3(0_f, vec2(0_f, 0_f)); WrapInFunction(expr); EXPECT_TRUE(r()->Resolve()) << r()->error(); auto* sem = Sem().Get(expr); EXPECT_NE(sem, nullptr); auto* vec = sem->Type()->As(); ASSERT_NE(vec, nullptr); EXPECT_TRUE(vec->type()->Is()); EXPECT_EQ(vec->Width(), 3u); EXPECT_TYPE(sem->ConstantValue()->Type(), sem->Type()); EXPECT_TRUE(sem->ConstantValue()->AllEqual()); EXPECT_TRUE(sem->ConstantValue()->AnyZero()); EXPECT_TRUE(sem->ConstantValue()->AllZero()); EXPECT_TRUE(sem->ConstantValue()->Index(0)->AllEqual()); EXPECT_TRUE(sem->ConstantValue()->Index(0)->AnyZero()); EXPECT_TRUE(sem->ConstantValue()->Index(0)->AllZero()); EXPECT_EQ(sem->ConstantValue()->Index(0)->ValueAs(), 0_f); EXPECT_TRUE(sem->ConstantValue()->Index(1)->AllEqual()); EXPECT_TRUE(sem->ConstantValue()->Index(1)->AnyZero()); EXPECT_TRUE(sem->ConstantValue()->Index(1)->AllZero()); EXPECT_EQ(sem->ConstantValue()->Index(1)->ValueAs(), 0_f); EXPECT_TRUE(sem->ConstantValue()->Index(2)->AllEqual()); EXPECT_TRUE(sem->ConstantValue()->Index(2)->AnyZero()); EXPECT_TRUE(sem->ConstantValue()->Index(2)->AllZero()); EXPECT_EQ(sem->ConstantValue()->Index(2)->ValueAs(), 0_f); } TEST_F(ResolverConstEvalTest, Vec3_MixConstruct_f32_all_negative_0) { auto* expr = vec3(vec2(-0_f, -0_f), -0_f); WrapInFunction(expr); EXPECT_TRUE(r()->Resolve()) << r()->error(); auto* sem = Sem().Get(expr); EXPECT_NE(sem, nullptr); auto* vec = sem->Type()->As(); ASSERT_NE(vec, nullptr); EXPECT_TRUE(vec->type()->Is()); EXPECT_EQ(vec->Width(), 3u); EXPECT_TYPE(sem->ConstantValue()->Type(), sem->Type()); EXPECT_TRUE(sem->ConstantValue()->AllEqual()); EXPECT_FALSE(sem->ConstantValue()->AnyZero()); EXPECT_FALSE(sem->ConstantValue()->AllZero()); EXPECT_TRUE(sem->ConstantValue()->Index(0)->AllEqual()); EXPECT_FALSE(sem->ConstantValue()->Index(0)->AnyZero()); EXPECT_FALSE(sem->ConstantValue()->Index(0)->AllZero()); EXPECT_EQ(sem->ConstantValue()->Index(0)->ValueAs(), -0_f); EXPECT_TRUE(sem->ConstantValue()->Index(1)->AllEqual()); EXPECT_FALSE(sem->ConstantValue()->Index(1)->AnyZero()); EXPECT_FALSE(sem->ConstantValue()->Index(1)->AllZero()); EXPECT_EQ(sem->ConstantValue()->Index(1)->ValueAs(), -0_f); EXPECT_TRUE(sem->ConstantValue()->Index(2)->AllEqual()); EXPECT_FALSE(sem->ConstantValue()->Index(2)->AnyZero()); EXPECT_FALSE(sem->ConstantValue()->Index(2)->AllZero()); EXPECT_EQ(sem->ConstantValue()->Index(2)->ValueAs(), -0_f); } TEST_F(ResolverConstEvalTest, Vec3_MixConstruct_f32_mixed_sign_0) { auto* expr = vec3(0_f, vec2(-0_f, 0_f)); WrapInFunction(expr); EXPECT_TRUE(r()->Resolve()) << r()->error(); auto* sem = Sem().Get(expr); EXPECT_NE(sem, nullptr); auto* vec = sem->Type()->As(); ASSERT_NE(vec, nullptr); EXPECT_TRUE(vec->type()->Is()); EXPECT_EQ(vec->Width(), 3u); EXPECT_TYPE(sem->ConstantValue()->Type(), sem->Type()); EXPECT_FALSE(sem->ConstantValue()->AllEqual()); EXPECT_TRUE(sem->ConstantValue()->AnyZero()); EXPECT_FALSE(sem->ConstantValue()->AllZero()); EXPECT_TRUE(sem->ConstantValue()->Index(0)->AllEqual()); EXPECT_TRUE(sem->ConstantValue()->Index(0)->AnyZero()); EXPECT_TRUE(sem->ConstantValue()->Index(0)->AllZero()); EXPECT_EQ(sem->ConstantValue()->Index(0)->ValueAs(), 0_f); EXPECT_TRUE(sem->ConstantValue()->Index(1)->AllEqual()); EXPECT_FALSE(sem->ConstantValue()->Index(1)->AnyZero()); EXPECT_FALSE(sem->ConstantValue()->Index(1)->AllZero()); EXPECT_EQ(sem->ConstantValue()->Index(1)->ValueAs(), -0_f); EXPECT_TRUE(sem->ConstantValue()->Index(2)->AllEqual()); EXPECT_TRUE(sem->ConstantValue()->Index(2)->AnyZero()); EXPECT_TRUE(sem->ConstantValue()->Index(2)->AllZero()); EXPECT_EQ(sem->ConstantValue()->Index(2)->ValueAs(), 0_f); } TEST_F(ResolverConstEvalTest, Vec3_MixConstruct_f16) { Enable(ast::Extension::kF16); auto* expr = vec3(1_h, vec2(2_h, 3_h)); WrapInFunction(expr); EXPECT_TRUE(r()->Resolve()) << r()->error(); auto* sem = Sem().Get(expr); EXPECT_NE(sem, nullptr); auto* vec = sem->Type()->As(); ASSERT_NE(vec, nullptr); EXPECT_TRUE(vec->type()->Is()); EXPECT_EQ(vec->Width(), 3u); EXPECT_TYPE(sem->ConstantValue()->Type(), sem->Type()); EXPECT_FALSE(sem->ConstantValue()->AllEqual()); EXPECT_FALSE(sem->ConstantValue()->AnyZero()); EXPECT_FALSE(sem->ConstantValue()->AllZero()); EXPECT_TRUE(sem->ConstantValue()->Index(0)->AllEqual()); EXPECT_FALSE(sem->ConstantValue()->Index(0)->AnyZero()); EXPECT_FALSE(sem->ConstantValue()->Index(0)->AllZero()); EXPECT_EQ(sem->ConstantValue()->Index(0)->ValueAs(), 1.f); EXPECT_TRUE(sem->ConstantValue()->Index(1)->AllEqual()); EXPECT_FALSE(sem->ConstantValue()->Index(1)->AnyZero()); EXPECT_FALSE(sem->ConstantValue()->Index(1)->AllZero()); EXPECT_EQ(sem->ConstantValue()->Index(1)->ValueAs(), 2.f); EXPECT_TRUE(sem->ConstantValue()->Index(2)->AllEqual()); EXPECT_FALSE(sem->ConstantValue()->Index(2)->AnyZero()); EXPECT_FALSE(sem->ConstantValue()->Index(2)->AllZero()); EXPECT_EQ(sem->ConstantValue()->Index(2)->ValueAs(), 3.f); } TEST_F(ResolverConstEvalTest, Vec3_MixConstruct_f16_all_10) { Enable(ast::Extension::kF16); auto* expr = vec3(10_h, vec2(10_h, 10_h)); WrapInFunction(expr); EXPECT_TRUE(r()->Resolve()) << r()->error(); auto* sem = Sem().Get(expr); EXPECT_NE(sem, nullptr); auto* vec = sem->Type()->As(); ASSERT_NE(vec, nullptr); EXPECT_TRUE(vec->type()->Is()); EXPECT_EQ(vec->Width(), 3u); EXPECT_TYPE(sem->ConstantValue()->Type(), sem->Type()); EXPECT_TRUE(sem->ConstantValue()->AllEqual()); EXPECT_FALSE(sem->ConstantValue()->AnyZero()); EXPECT_FALSE(sem->ConstantValue()->AllZero()); EXPECT_TRUE(sem->ConstantValue()->Index(0)->AllEqual()); EXPECT_FALSE(sem->ConstantValue()->Index(0)->AnyZero()); EXPECT_FALSE(sem->ConstantValue()->Index(0)->AllZero()); EXPECT_EQ(sem->ConstantValue()->Index(0)->ValueAs(), 10_h); EXPECT_TRUE(sem->ConstantValue()->Index(1)->AllEqual()); EXPECT_FALSE(sem->ConstantValue()->Index(1)->AnyZero()); EXPECT_FALSE(sem->ConstantValue()->Index(1)->AllZero()); EXPECT_EQ(sem->ConstantValue()->Index(1)->ValueAs(), 10_h); EXPECT_TRUE(sem->ConstantValue()->Index(2)->AllEqual()); EXPECT_FALSE(sem->ConstantValue()->Index(2)->AnyZero()); EXPECT_FALSE(sem->ConstantValue()->Index(2)->AllZero()); EXPECT_EQ(sem->ConstantValue()->Index(2)->ValueAs(), 10_h); } TEST_F(ResolverConstEvalTest, Vec3_MixConstruct_f16_all_positive_0) { Enable(ast::Extension::kF16); auto* expr = vec3(0_h, vec2(0_h, 0_h)); WrapInFunction(expr); EXPECT_TRUE(r()->Resolve()) << r()->error(); auto* sem = Sem().Get(expr); EXPECT_NE(sem, nullptr); auto* vec = sem->Type()->As(); ASSERT_NE(vec, nullptr); EXPECT_TRUE(vec->type()->Is()); EXPECT_EQ(vec->Width(), 3u); EXPECT_TYPE(sem->ConstantValue()->Type(), sem->Type()); EXPECT_TRUE(sem->ConstantValue()->AllEqual()); EXPECT_TRUE(sem->ConstantValue()->AnyZero()); EXPECT_TRUE(sem->ConstantValue()->AllZero()); EXPECT_TRUE(sem->ConstantValue()->Index(0)->AllEqual()); EXPECT_TRUE(sem->ConstantValue()->Index(0)->AnyZero()); EXPECT_TRUE(sem->ConstantValue()->Index(0)->AllZero()); EXPECT_EQ(sem->ConstantValue()->Index(0)->ValueAs(), 0_h); EXPECT_TRUE(sem->ConstantValue()->Index(1)->AllEqual()); EXPECT_TRUE(sem->ConstantValue()->Index(1)->AnyZero()); EXPECT_TRUE(sem->ConstantValue()->Index(1)->AllZero()); EXPECT_EQ(sem->ConstantValue()->Index(1)->ValueAs(), 0_h); EXPECT_TRUE(sem->ConstantValue()->Index(2)->AllEqual()); EXPECT_TRUE(sem->ConstantValue()->Index(2)->AnyZero()); EXPECT_TRUE(sem->ConstantValue()->Index(2)->AllZero()); EXPECT_EQ(sem->ConstantValue()->Index(2)->ValueAs(), 0_h); } TEST_F(ResolverConstEvalTest, Vec3_MixConstruct_f16_all_negative_0) { Enable(ast::Extension::kF16); auto* expr = vec3(vec2(-0_h, -0_h), -0_h); WrapInFunction(expr); EXPECT_TRUE(r()->Resolve()) << r()->error(); auto* sem = Sem().Get(expr); EXPECT_NE(sem, nullptr); auto* vec = sem->Type()->As(); ASSERT_NE(vec, nullptr); EXPECT_TRUE(vec->type()->Is()); EXPECT_EQ(vec->Width(), 3u); EXPECT_TYPE(sem->ConstantValue()->Type(), sem->Type()); EXPECT_TRUE(sem->ConstantValue()->AllEqual()); EXPECT_FALSE(sem->ConstantValue()->AnyZero()); EXPECT_FALSE(sem->ConstantValue()->AllZero()); EXPECT_TRUE(sem->ConstantValue()->Index(0)->AllEqual()); EXPECT_FALSE(sem->ConstantValue()->Index(0)->AnyZero()); EXPECT_FALSE(sem->ConstantValue()->Index(0)->AllZero()); EXPECT_EQ(sem->ConstantValue()->Index(0)->ValueAs(), -0_h); EXPECT_TRUE(sem->ConstantValue()->Index(1)->AllEqual()); EXPECT_FALSE(sem->ConstantValue()->Index(1)->AnyZero()); EXPECT_FALSE(sem->ConstantValue()->Index(1)->AllZero()); EXPECT_EQ(sem->ConstantValue()->Index(1)->ValueAs(), -0_h); EXPECT_TRUE(sem->ConstantValue()->Index(2)->AllEqual()); EXPECT_FALSE(sem->ConstantValue()->Index(2)->AnyZero()); EXPECT_FALSE(sem->ConstantValue()->Index(2)->AllZero()); EXPECT_EQ(sem->ConstantValue()->Index(2)->ValueAs(), -0_h); } TEST_F(ResolverConstEvalTest, Vec3_MixConstruct_f16_mixed_sign_0) { Enable(ast::Extension::kF16); auto* expr = vec3(0_h, vec2(-0_h, 0_h)); WrapInFunction(expr); EXPECT_TRUE(r()->Resolve()) << r()->error(); auto* sem = Sem().Get(expr); EXPECT_NE(sem, nullptr); auto* vec = sem->Type()->As(); ASSERT_NE(vec, nullptr); EXPECT_TRUE(vec->type()->Is()); EXPECT_EQ(vec->Width(), 3u); EXPECT_TYPE(sem->ConstantValue()->Type(), sem->Type()); EXPECT_FALSE(sem->ConstantValue()->AllEqual()); EXPECT_TRUE(sem->ConstantValue()->AnyZero()); EXPECT_FALSE(sem->ConstantValue()->AllZero()); EXPECT_TRUE(sem->ConstantValue()->Index(0)->AllEqual()); EXPECT_TRUE(sem->ConstantValue()->Index(0)->AnyZero()); EXPECT_TRUE(sem->ConstantValue()->Index(0)->AllZero()); EXPECT_EQ(sem->ConstantValue()->Index(0)->ValueAs(), 0_h); EXPECT_TRUE(sem->ConstantValue()->Index(1)->AllEqual()); EXPECT_FALSE(sem->ConstantValue()->Index(1)->AnyZero()); EXPECT_FALSE(sem->ConstantValue()->Index(1)->AllZero()); EXPECT_EQ(sem->ConstantValue()->Index(1)->ValueAs(), -0_h); EXPECT_TRUE(sem->ConstantValue()->Index(2)->AllEqual()); EXPECT_TRUE(sem->ConstantValue()->Index(2)->AnyZero()); EXPECT_TRUE(sem->ConstantValue()->Index(2)->AllZero()); EXPECT_EQ(sem->ConstantValue()->Index(2)->ValueAs(), 0_h); } TEST_F(ResolverConstEvalTest, Vec3_MixConstruct_bool) { auto* expr = vec3(vec2(true, false), true); WrapInFunction(expr); EXPECT_TRUE(r()->Resolve()) << r()->error(); auto* sem = Sem().Get(expr); ASSERT_NE(sem, nullptr); auto* vec = sem->Type()->As(); ASSERT_NE(vec, nullptr); EXPECT_TRUE(vec->type()->Is()); EXPECT_EQ(vec->Width(), 3u); EXPECT_TYPE(sem->ConstantValue()->Type(), sem->Type()); EXPECT_FALSE(sem->ConstantValue()->AllEqual()); EXPECT_TRUE(sem->ConstantValue()->AnyZero()); EXPECT_FALSE(sem->ConstantValue()->AllZero()); EXPECT_TRUE(sem->ConstantValue()->Index(0)->AllEqual()); EXPECT_FALSE(sem->ConstantValue()->Index(0)->AnyZero()); EXPECT_FALSE(sem->ConstantValue()->Index(0)->AllZero()); EXPECT_EQ(sem->ConstantValue()->Index(0)->ValueAs(), true); EXPECT_TRUE(sem->ConstantValue()->Index(1)->AllEqual()); EXPECT_TRUE(sem->ConstantValue()->Index(1)->AnyZero()); EXPECT_TRUE(sem->ConstantValue()->Index(1)->AllZero()); EXPECT_EQ(sem->ConstantValue()->Index(1)->ValueAs(), false); EXPECT_TRUE(sem->ConstantValue()->Index(2)->AllEqual()); EXPECT_FALSE(sem->ConstantValue()->Index(2)->AnyZero()); EXPECT_FALSE(sem->ConstantValue()->Index(2)->AllZero()); EXPECT_EQ(sem->ConstantValue()->Index(2)->ValueAs(), true); } TEST_F(ResolverConstEvalTest, Vec3_MixConstruct_all_true) { auto* expr = vec3(true, vec2(true, true)); WrapInFunction(expr); EXPECT_TRUE(r()->Resolve()) << r()->error(); auto* sem = Sem().Get(expr); ASSERT_NE(sem, nullptr); auto* vec = sem->Type()->As(); ASSERT_NE(vec, nullptr); EXPECT_TRUE(vec->type()->Is()); EXPECT_EQ(vec->Width(), 3u); EXPECT_TYPE(sem->ConstantValue()->Type(), sem->Type()); EXPECT_TRUE(sem->ConstantValue()->AllEqual()); EXPECT_FALSE(sem->ConstantValue()->AnyZero()); EXPECT_FALSE(sem->ConstantValue()->AllZero()); EXPECT_TRUE(sem->ConstantValue()->Index(0)->AllEqual()); EXPECT_FALSE(sem->ConstantValue()->Index(0)->AnyZero()); EXPECT_FALSE(sem->ConstantValue()->Index(0)->AllZero()); EXPECT_EQ(sem->ConstantValue()->Index(0)->ValueAs(), true); EXPECT_TRUE(sem->ConstantValue()->Index(1)->AllEqual()); EXPECT_FALSE(sem->ConstantValue()->Index(1)->AnyZero()); EXPECT_FALSE(sem->ConstantValue()->Index(1)->AllZero()); EXPECT_EQ(sem->ConstantValue()->Index(1)->ValueAs(), true); EXPECT_TRUE(sem->ConstantValue()->Index(2)->AllEqual()); EXPECT_FALSE(sem->ConstantValue()->Index(2)->AnyZero()); EXPECT_FALSE(sem->ConstantValue()->Index(2)->AllZero()); EXPECT_EQ(sem->ConstantValue()->Index(2)->ValueAs(), true); } TEST_F(ResolverConstEvalTest, Vec3_MixConstruct_all_false) { auto* expr = vec3(false, vec2(false, false)); WrapInFunction(expr); EXPECT_TRUE(r()->Resolve()) << r()->error(); auto* sem = Sem().Get(expr); ASSERT_NE(sem, nullptr); auto* vec = sem->Type()->As(); ASSERT_NE(vec, nullptr); EXPECT_TRUE(vec->type()->Is()); EXPECT_EQ(vec->Width(), 3u); EXPECT_TYPE(sem->ConstantValue()->Type(), sem->Type()); EXPECT_TRUE(sem->ConstantValue()->AllEqual()); EXPECT_TRUE(sem->ConstantValue()->AnyZero()); EXPECT_TRUE(sem->ConstantValue()->AllZero()); EXPECT_TRUE(sem->ConstantValue()->Index(0)->AllEqual()); EXPECT_TRUE(sem->ConstantValue()->Index(0)->AnyZero()); EXPECT_TRUE(sem->ConstantValue()->Index(0)->AllZero()); EXPECT_EQ(sem->ConstantValue()->Index(0)->ValueAs(), false); EXPECT_TRUE(sem->ConstantValue()->Index(1)->AllEqual()); EXPECT_TRUE(sem->ConstantValue()->Index(1)->AnyZero()); EXPECT_TRUE(sem->ConstantValue()->Index(1)->AllZero()); EXPECT_EQ(sem->ConstantValue()->Index(1)->ValueAs(), false); EXPECT_TRUE(sem->ConstantValue()->Index(2)->AllEqual()); EXPECT_TRUE(sem->ConstantValue()->Index(2)->AnyZero()); EXPECT_TRUE(sem->ConstantValue()->Index(2)->AllZero()); EXPECT_EQ(sem->ConstantValue()->Index(2)->ValueAs(), false); } TEST_F(ResolverConstEvalTest, Mat2x3_ZeroInit_f32) { auto* expr = mat2x3(); WrapInFunction(expr); EXPECT_TRUE(r()->Resolve()) << r()->error(); auto* sem = Sem().Get(expr); ASSERT_NE(sem, nullptr); auto* mat = sem->Type()->As(); ASSERT_NE(mat, nullptr); EXPECT_TRUE(mat->type()->Is()); EXPECT_EQ(mat->columns(), 2u); EXPECT_EQ(mat->rows(), 3u); EXPECT_TYPE(sem->ConstantValue()->Type(), sem->Type()); EXPECT_TRUE(sem->ConstantValue()->AllEqual()); EXPECT_TRUE(sem->ConstantValue()->AnyZero()); EXPECT_TRUE(sem->ConstantValue()->AllZero()); EXPECT_TRUE(sem->ConstantValue()->Index(0)->Index(0)->AllEqual()); EXPECT_TRUE(sem->ConstantValue()->Index(0)->Index(0)->AnyZero()); EXPECT_TRUE(sem->ConstantValue()->Index(0)->Index(0)->AllZero()); EXPECT_EQ(sem->ConstantValue()->Index(0)->Index(0)->ValueAs(), 0._f); EXPECT_TRUE(sem->ConstantValue()->Index(0)->Index(1)->AllEqual()); EXPECT_TRUE(sem->ConstantValue()->Index(0)->Index(1)->AnyZero()); EXPECT_TRUE(sem->ConstantValue()->Index(0)->Index(1)->AllZero()); EXPECT_EQ(sem->ConstantValue()->Index(0)->Index(1)->ValueAs(), 0._f); EXPECT_TRUE(sem->ConstantValue()->Index(0)->Index(2)->AllEqual()); EXPECT_TRUE(sem->ConstantValue()->Index(0)->Index(2)->AnyZero()); EXPECT_TRUE(sem->ConstantValue()->Index(0)->Index(2)->AllZero()); EXPECT_EQ(sem->ConstantValue()->Index(0)->Index(2)->ValueAs(), 0._f); EXPECT_TRUE(sem->ConstantValue()->Index(1)->Index(0)->AllEqual()); EXPECT_TRUE(sem->ConstantValue()->Index(1)->Index(0)->AnyZero()); EXPECT_TRUE(sem->ConstantValue()->Index(1)->Index(0)->AllZero()); EXPECT_EQ(sem->ConstantValue()->Index(1)->Index(0)->ValueAs(), 0._f); EXPECT_TRUE(sem->ConstantValue()->Index(1)->Index(1)->AllEqual()); EXPECT_TRUE(sem->ConstantValue()->Index(1)->Index(1)->AnyZero()); EXPECT_TRUE(sem->ConstantValue()->Index(1)->Index(1)->AllZero()); EXPECT_EQ(sem->ConstantValue()->Index(1)->Index(1)->ValueAs(), 0._f); EXPECT_TRUE(sem->ConstantValue()->Index(1)->Index(2)->AllEqual()); EXPECT_TRUE(sem->ConstantValue()->Index(1)->Index(2)->AnyZero()); EXPECT_TRUE(sem->ConstantValue()->Index(1)->Index(2)->AllZero()); EXPECT_EQ(sem->ConstantValue()->Index(1)->Index(2)->ValueAs(), 0._f); } TEST_F(ResolverConstEvalTest, Mat2x3_ZeroInit_f16) { Enable(ast::Extension::kF16); auto* expr = mat2x3(); WrapInFunction(expr); EXPECT_TRUE(r()->Resolve()) << r()->error(); auto* sem = Sem().Get(expr); EXPECT_NE(sem, nullptr); auto* mat = sem->Type()->As(); ASSERT_NE(mat, nullptr); EXPECT_TRUE(mat->type()->Is()); EXPECT_EQ(mat->columns(), 2u); EXPECT_EQ(mat->rows(), 3u); EXPECT_TYPE(sem->ConstantValue()->Type(), sem->Type()); EXPECT_TRUE(sem->ConstantValue()->AllEqual()); EXPECT_TRUE(sem->ConstantValue()->AnyZero()); EXPECT_TRUE(sem->ConstantValue()->AllZero()); EXPECT_TRUE(sem->ConstantValue()->Index(0)->Index(0)->AllEqual()); EXPECT_TRUE(sem->ConstantValue()->Index(0)->Index(0)->AnyZero()); EXPECT_TRUE(sem->ConstantValue()->Index(0)->Index(0)->AllZero()); EXPECT_EQ(sem->ConstantValue()->Index(0)->Index(0)->ValueAs(), 0._h); EXPECT_TRUE(sem->ConstantValue()->Index(0)->Index(1)->AllEqual()); EXPECT_TRUE(sem->ConstantValue()->Index(0)->Index(1)->AnyZero()); EXPECT_TRUE(sem->ConstantValue()->Index(0)->Index(1)->AllZero()); EXPECT_EQ(sem->ConstantValue()->Index(0)->Index(1)->ValueAs(), 0._h); EXPECT_TRUE(sem->ConstantValue()->Index(0)->Index(2)->AllEqual()); EXPECT_TRUE(sem->ConstantValue()->Index(0)->Index(2)->AnyZero()); EXPECT_TRUE(sem->ConstantValue()->Index(0)->Index(2)->AllZero()); EXPECT_EQ(sem->ConstantValue()->Index(0)->Index(2)->ValueAs(), 0._h); EXPECT_TRUE(sem->ConstantValue()->Index(1)->Index(0)->AllEqual()); EXPECT_TRUE(sem->ConstantValue()->Index(1)->Index(0)->AnyZero()); EXPECT_TRUE(sem->ConstantValue()->Index(1)->Index(0)->AllZero()); EXPECT_EQ(sem->ConstantValue()->Index(1)->Index(0)->ValueAs(), 0._h); EXPECT_TRUE(sem->ConstantValue()->Index(1)->Index(1)->AllEqual()); EXPECT_TRUE(sem->ConstantValue()->Index(1)->Index(1)->AnyZero()); EXPECT_TRUE(sem->ConstantValue()->Index(1)->Index(1)->AllZero()); EXPECT_EQ(sem->ConstantValue()->Index(1)->Index(1)->ValueAs(), 0._h); EXPECT_TRUE(sem->ConstantValue()->Index(1)->Index(2)->AllEqual()); EXPECT_TRUE(sem->ConstantValue()->Index(1)->Index(2)->AnyZero()); EXPECT_TRUE(sem->ConstantValue()->Index(1)->Index(2)->AllZero()); EXPECT_EQ(sem->ConstantValue()->Index(1)->Index(2)->ValueAs(), 0._h); } TEST_F(ResolverConstEvalTest, Mat3x2_Construct_Scalars_af) { auto* expr = Construct(ty.mat(nullptr, 3, 2), 1.0_a, 2.0_a, 3.0_a, 4.0_a, 5.0_a, 6.0_a); WrapInFunction(expr); EXPECT_TRUE(r()->Resolve()) << r()->error(); auto* sem = Sem().Get(expr); ASSERT_NE(sem, nullptr); auto* mat = sem->Type()->As(); ASSERT_NE(mat, nullptr); EXPECT_TRUE(mat->type()->Is()); EXPECT_EQ(mat->columns(), 3u); EXPECT_EQ(mat->rows(), 2u); EXPECT_TYPE(sem->ConstantValue()->Type(), sem->Type()); EXPECT_FALSE(sem->ConstantValue()->AllEqual()); EXPECT_FALSE(sem->ConstantValue()->AnyZero()); EXPECT_FALSE(sem->ConstantValue()->AllZero()); EXPECT_TRUE(sem->ConstantValue()->Index(0)->Index(0)->AllEqual()); EXPECT_FALSE(sem->ConstantValue()->Index(0)->Index(0)->AnyZero()); EXPECT_FALSE(sem->ConstantValue()->Index(0)->Index(0)->AllZero()); EXPECT_EQ(sem->ConstantValue()->Index(0)->Index(0)->ValueAs(), 1._a); EXPECT_TRUE(sem->ConstantValue()->Index(0)->Index(1)->AllEqual()); EXPECT_FALSE(sem->ConstantValue()->Index(0)->Index(1)->AnyZero()); EXPECT_FALSE(sem->ConstantValue()->Index(0)->Index(1)->AllZero()); EXPECT_EQ(sem->ConstantValue()->Index(0)->Index(1)->ValueAs(), 2._a); EXPECT_TRUE(sem->ConstantValue()->Index(1)->Index(0)->AllEqual()); EXPECT_FALSE(sem->ConstantValue()->Index(1)->Index(0)->AnyZero()); EXPECT_FALSE(sem->ConstantValue()->Index(1)->Index(0)->AllZero()); EXPECT_EQ(sem->ConstantValue()->Index(1)->Index(0)->ValueAs(), 3._a); EXPECT_TRUE(sem->ConstantValue()->Index(1)->Index(1)->AllEqual()); EXPECT_FALSE(sem->ConstantValue()->Index(1)->Index(1)->AnyZero()); EXPECT_FALSE(sem->ConstantValue()->Index(1)->Index(1)->AllZero()); EXPECT_EQ(sem->ConstantValue()->Index(1)->Index(1)->ValueAs(), 4._a); EXPECT_TRUE(sem->ConstantValue()->Index(2)->Index(0)->AllEqual()); EXPECT_FALSE(sem->ConstantValue()->Index(2)->Index(0)->AnyZero()); EXPECT_FALSE(sem->ConstantValue()->Index(2)->Index(0)->AllZero()); EXPECT_EQ(sem->ConstantValue()->Index(2)->Index(0)->ValueAs(), 5._a); EXPECT_TRUE(sem->ConstantValue()->Index(2)->Index(1)->AllEqual()); EXPECT_FALSE(sem->ConstantValue()->Index(2)->Index(1)->AnyZero()); EXPECT_FALSE(sem->ConstantValue()->Index(2)->Index(1)->AllZero()); EXPECT_EQ(sem->ConstantValue()->Index(2)->Index(1)->ValueAs(), 6._a); } TEST_F(ResolverConstEvalTest, Mat3x2_Construct_Columns_af) { auto* expr = Construct(ty.mat(nullptr, 3, 2), // vec(nullptr, 2u, 1.0_a, 2.0_a), // vec(nullptr, 2u, 3.0_a, 4.0_a), // vec(nullptr, 2u, 5.0_a, 6.0_a)); WrapInFunction(expr); EXPECT_TRUE(r()->Resolve()) << r()->error(); auto* sem = Sem().Get(expr); ASSERT_NE(sem, nullptr); auto* mat = sem->Type()->As(); ASSERT_NE(mat, nullptr); EXPECT_TRUE(mat->type()->Is()); EXPECT_EQ(mat->columns(), 3u); EXPECT_EQ(mat->rows(), 2u); EXPECT_TYPE(sem->ConstantValue()->Type(), sem->Type()); EXPECT_FALSE(sem->ConstantValue()->AllEqual()); EXPECT_FALSE(sem->ConstantValue()->AnyZero()); EXPECT_FALSE(sem->ConstantValue()->AllZero()); EXPECT_TRUE(sem->ConstantValue()->Index(0)->Index(0)->AllEqual()); EXPECT_FALSE(sem->ConstantValue()->Index(0)->Index(0)->AnyZero()); EXPECT_FALSE(sem->ConstantValue()->Index(0)->Index(0)->AllZero()); EXPECT_EQ(sem->ConstantValue()->Index(0)->Index(0)->ValueAs(), 1._a); EXPECT_TRUE(sem->ConstantValue()->Index(0)->Index(1)->AllEqual()); EXPECT_FALSE(sem->ConstantValue()->Index(0)->Index(1)->AnyZero()); EXPECT_FALSE(sem->ConstantValue()->Index(0)->Index(1)->AllZero()); EXPECT_EQ(sem->ConstantValue()->Index(0)->Index(1)->ValueAs(), 2._a); EXPECT_TRUE(sem->ConstantValue()->Index(1)->Index(0)->AllEqual()); EXPECT_FALSE(sem->ConstantValue()->Index(1)->Index(0)->AnyZero()); EXPECT_FALSE(sem->ConstantValue()->Index(1)->Index(0)->AllZero()); EXPECT_EQ(sem->ConstantValue()->Index(1)->Index(0)->ValueAs(), 3._a); EXPECT_TRUE(sem->ConstantValue()->Index(1)->Index(1)->AllEqual()); EXPECT_FALSE(sem->ConstantValue()->Index(1)->Index(1)->AnyZero()); EXPECT_FALSE(sem->ConstantValue()->Index(1)->Index(1)->AllZero()); EXPECT_EQ(sem->ConstantValue()->Index(1)->Index(1)->ValueAs(), 4._a); EXPECT_TRUE(sem->ConstantValue()->Index(2)->Index(0)->AllEqual()); EXPECT_FALSE(sem->ConstantValue()->Index(2)->Index(0)->AnyZero()); EXPECT_FALSE(sem->ConstantValue()->Index(2)->Index(0)->AllZero()); EXPECT_EQ(sem->ConstantValue()->Index(2)->Index(0)->ValueAs(), 5._a); EXPECT_TRUE(sem->ConstantValue()->Index(2)->Index(1)->AllEqual()); EXPECT_FALSE(sem->ConstantValue()->Index(2)->Index(1)->AnyZero()); EXPECT_FALSE(sem->ConstantValue()->Index(2)->Index(1)->AllZero()); EXPECT_EQ(sem->ConstantValue()->Index(2)->Index(1)->ValueAs(), 6._a); } TEST_F(ResolverConstEvalTest, Array_i32_Zero) { auto* expr = Construct(ty.array()); WrapInFunction(expr); EXPECT_TRUE(r()->Resolve()) << r()->error(); auto* sem = Sem().Get(expr); ASSERT_NE(sem, nullptr); auto* arr = sem->Type()->As(); ASSERT_NE(arr, nullptr); EXPECT_TRUE(arr->ElemType()->Is()); EXPECT_TYPE(sem->ConstantValue()->Type(), sem->Type()); EXPECT_TRUE(sem->ConstantValue()->AllEqual()); EXPECT_TRUE(sem->ConstantValue()->AnyZero()); EXPECT_TRUE(sem->ConstantValue()->AllZero()); EXPECT_TRUE(sem->ConstantValue()->Index(0)->AllEqual()); EXPECT_TRUE(sem->ConstantValue()->Index(0)->AnyZero()); EXPECT_TRUE(sem->ConstantValue()->Index(0)->AllZero()); EXPECT_EQ(sem->ConstantValue()->Index(0)->ValueAs(), 0_i); EXPECT_TRUE(sem->ConstantValue()->Index(1)->AllEqual()); EXPECT_TRUE(sem->ConstantValue()->Index(1)->AnyZero()); EXPECT_TRUE(sem->ConstantValue()->Index(1)->AllZero()); EXPECT_EQ(sem->ConstantValue()->Index(1)->ValueAs(), 0_i); EXPECT_TRUE(sem->ConstantValue()->Index(2)->AllEqual()); EXPECT_TRUE(sem->ConstantValue()->Index(2)->AnyZero()); EXPECT_TRUE(sem->ConstantValue()->Index(2)->AllZero()); EXPECT_EQ(sem->ConstantValue()->Index(2)->ValueAs(), 0_i); EXPECT_TRUE(sem->ConstantValue()->Index(3)->AllEqual()); EXPECT_TRUE(sem->ConstantValue()->Index(3)->AnyZero()); EXPECT_TRUE(sem->ConstantValue()->Index(3)->AllZero()); EXPECT_EQ(sem->ConstantValue()->Index(3)->ValueAs(), 0_i); } TEST_F(ResolverConstEvalTest, Array_f32_Zero) { auto* expr = Construct(ty.array()); WrapInFunction(expr); EXPECT_TRUE(r()->Resolve()) << r()->error(); auto* sem = Sem().Get(expr); ASSERT_NE(sem, nullptr); auto* arr = sem->Type()->As(); ASSERT_NE(arr, nullptr); EXPECT_TRUE(arr->ElemType()->Is()); EXPECT_TYPE(sem->ConstantValue()->Type(), sem->Type()); EXPECT_TRUE(sem->ConstantValue()->AllEqual()); EXPECT_TRUE(sem->ConstantValue()->AnyZero()); EXPECT_TRUE(sem->ConstantValue()->AllZero()); EXPECT_TRUE(sem->ConstantValue()->Index(0)->AllEqual()); EXPECT_TRUE(sem->ConstantValue()->Index(0)->AnyZero()); EXPECT_TRUE(sem->ConstantValue()->Index(0)->AllZero()); EXPECT_EQ(sem->ConstantValue()->Index(0)->ValueAs(), 0_f); EXPECT_TRUE(sem->ConstantValue()->Index(1)->AllEqual()); EXPECT_TRUE(sem->ConstantValue()->Index(1)->AnyZero()); EXPECT_TRUE(sem->ConstantValue()->Index(1)->AllZero()); EXPECT_EQ(sem->ConstantValue()->Index(1)->ValueAs(), 0_f); EXPECT_TRUE(sem->ConstantValue()->Index(2)->AllEqual()); EXPECT_TRUE(sem->ConstantValue()->Index(2)->AnyZero()); EXPECT_TRUE(sem->ConstantValue()->Index(2)->AllZero()); EXPECT_EQ(sem->ConstantValue()->Index(2)->ValueAs(), 0_f); EXPECT_TRUE(sem->ConstantValue()->Index(3)->AllEqual()); EXPECT_TRUE(sem->ConstantValue()->Index(3)->AnyZero()); EXPECT_TRUE(sem->ConstantValue()->Index(3)->AllZero()); EXPECT_EQ(sem->ConstantValue()->Index(3)->ValueAs(), 0_f); } TEST_F(ResolverConstEvalTest, Array_vec3_f32_Zero) { auto* expr = Construct(ty.array(ty.vec3(), 2_u)); WrapInFunction(expr); EXPECT_TRUE(r()->Resolve()) << r()->error(); auto* sem = Sem().Get(expr); ASSERT_NE(sem, nullptr); auto* arr = sem->Type()->As(); ASSERT_NE(arr, nullptr); EXPECT_TRUE(arr->ElemType()->Is()); EXPECT_TYPE(sem->ConstantValue()->Type(), sem->Type()); EXPECT_TRUE(sem->ConstantValue()->AllEqual()); EXPECT_TRUE(sem->ConstantValue()->AnyZero()); EXPECT_TRUE(sem->ConstantValue()->AllZero()); EXPECT_TRUE(sem->ConstantValue()->Index(0)->Index(0)->AllEqual()); EXPECT_TRUE(sem->ConstantValue()->Index(0)->Index(0)->AnyZero()); EXPECT_TRUE(sem->ConstantValue()->Index(0)->Index(0)->AllZero()); EXPECT_EQ(sem->ConstantValue()->Index(0)->Index(0)->ValueAs(), 0_f); EXPECT_TRUE(sem->ConstantValue()->Index(0)->Index(1)->AllEqual()); EXPECT_TRUE(sem->ConstantValue()->Index(0)->Index(1)->AnyZero()); EXPECT_TRUE(sem->ConstantValue()->Index(0)->Index(1)->AllZero()); EXPECT_EQ(sem->ConstantValue()->Index(0)->Index(1)->ValueAs(), 0_f); EXPECT_TRUE(sem->ConstantValue()->Index(0)->Index(2)->AllEqual()); EXPECT_TRUE(sem->ConstantValue()->Index(0)->Index(2)->AnyZero()); EXPECT_TRUE(sem->ConstantValue()->Index(0)->Index(2)->AllZero()); EXPECT_EQ(sem->ConstantValue()->Index(0)->Index(2)->ValueAs(), 0_f); EXPECT_TRUE(sem->ConstantValue()->Index(1)->Index(0)->AllEqual()); EXPECT_TRUE(sem->ConstantValue()->Index(1)->Index(0)->AnyZero()); EXPECT_TRUE(sem->ConstantValue()->Index(1)->Index(0)->AllZero()); EXPECT_EQ(sem->ConstantValue()->Index(1)->Index(0)->ValueAs(), 0_f); EXPECT_TRUE(sem->ConstantValue()->Index(1)->Index(1)->AllEqual()); EXPECT_TRUE(sem->ConstantValue()->Index(1)->Index(1)->AnyZero()); EXPECT_TRUE(sem->ConstantValue()->Index(1)->Index(1)->AllZero()); EXPECT_EQ(sem->ConstantValue()->Index(1)->Index(1)->ValueAs(), 0_f); EXPECT_TRUE(sem->ConstantValue()->Index(1)->Index(2)->AllEqual()); EXPECT_TRUE(sem->ConstantValue()->Index(1)->Index(2)->AnyZero()); EXPECT_TRUE(sem->ConstantValue()->Index(1)->Index(2)->AllZero()); EXPECT_EQ(sem->ConstantValue()->Index(1)->Index(2)->ValueAs(), 0_f); } TEST_F(ResolverConstEvalTest, Array_Struct_f32_Zero) { Structure("S", utils::Vector{ Member("m1", ty.f32()), Member("m2", ty.f32()), }); auto* expr = Construct(ty.array(ty.type_name("S"), 2_u)); WrapInFunction(expr); EXPECT_TRUE(r()->Resolve()) << r()->error(); auto* sem = Sem().Get(expr); ASSERT_NE(sem, nullptr); auto* arr = sem->Type()->As(); ASSERT_NE(arr, nullptr); EXPECT_TRUE(arr->ElemType()->Is()); EXPECT_TYPE(sem->ConstantValue()->Type(), sem->Type()); EXPECT_TRUE(sem->ConstantValue()->AllEqual()); EXPECT_TRUE(sem->ConstantValue()->AnyZero()); EXPECT_TRUE(sem->ConstantValue()->AllZero()); EXPECT_TRUE(sem->ConstantValue()->Index(0)->Index(0)->AllEqual()); EXPECT_TRUE(sem->ConstantValue()->Index(0)->Index(0)->AnyZero()); EXPECT_TRUE(sem->ConstantValue()->Index(0)->Index(0)->AllZero()); EXPECT_EQ(sem->ConstantValue()->Index(0)->Index(0)->ValueAs(), 0_f); EXPECT_TRUE(sem->ConstantValue()->Index(0)->Index(1)->AllEqual()); EXPECT_TRUE(sem->ConstantValue()->Index(0)->Index(1)->AnyZero()); EXPECT_TRUE(sem->ConstantValue()->Index(0)->Index(1)->AllZero()); EXPECT_EQ(sem->ConstantValue()->Index(0)->Index(1)->ValueAs(), 0_f); EXPECT_TRUE(sem->ConstantValue()->Index(1)->Index(0)->AllEqual()); EXPECT_TRUE(sem->ConstantValue()->Index(1)->Index(0)->AnyZero()); EXPECT_TRUE(sem->ConstantValue()->Index(1)->Index(0)->AllZero()); EXPECT_EQ(sem->ConstantValue()->Index(1)->Index(0)->ValueAs(), 0_f); EXPECT_TRUE(sem->ConstantValue()->Index(1)->Index(1)->AllEqual()); EXPECT_TRUE(sem->ConstantValue()->Index(1)->Index(1)->AnyZero()); EXPECT_TRUE(sem->ConstantValue()->Index(1)->Index(1)->AllZero()); EXPECT_EQ(sem->ConstantValue()->Index(1)->Index(1)->ValueAs(), 0_f); } TEST_F(ResolverConstEvalTest, Array_i32_Elements) { auto* expr = Construct(ty.array(), 10_i, 20_i, 30_i, 40_i); WrapInFunction(expr); EXPECT_TRUE(r()->Resolve()) << r()->error(); auto* sem = Sem().Get(expr); ASSERT_NE(sem, nullptr); auto* arr = sem->Type()->As(); ASSERT_NE(arr, nullptr); EXPECT_TRUE(arr->ElemType()->Is()); EXPECT_TYPE(sem->ConstantValue()->Type(), sem->Type()); EXPECT_FALSE(sem->ConstantValue()->AllEqual()); EXPECT_FALSE(sem->ConstantValue()->AnyZero()); EXPECT_FALSE(sem->ConstantValue()->AllZero()); EXPECT_TRUE(sem->ConstantValue()->Index(0)->AllEqual()); EXPECT_FALSE(sem->ConstantValue()->Index(0)->AnyZero()); EXPECT_FALSE(sem->ConstantValue()->Index(0)->AllZero()); EXPECT_EQ(sem->ConstantValue()->Index(0)->ValueAs(), 10_i); EXPECT_TRUE(sem->ConstantValue()->Index(1)->AllEqual()); EXPECT_FALSE(sem->ConstantValue()->Index(1)->AnyZero()); EXPECT_FALSE(sem->ConstantValue()->Index(1)->AllZero()); EXPECT_EQ(sem->ConstantValue()->Index(1)->ValueAs(), 20_i); EXPECT_TRUE(sem->ConstantValue()->Index(2)->AllEqual()); EXPECT_FALSE(sem->ConstantValue()->Index(2)->AnyZero()); EXPECT_FALSE(sem->ConstantValue()->Index(2)->AllZero()); EXPECT_EQ(sem->ConstantValue()->Index(2)->ValueAs(), 30_i); EXPECT_TRUE(sem->ConstantValue()->Index(3)->AllEqual()); EXPECT_FALSE(sem->ConstantValue()->Index(3)->AnyZero()); EXPECT_FALSE(sem->ConstantValue()->Index(3)->AllZero()); EXPECT_EQ(sem->ConstantValue()->Index(3)->ValueAs(), 40_i); } namespace ArrayInit { struct Case { Value input; }; static Case C(Value input) { return Case{std::move(input)}; } static std::ostream& operator<<(std::ostream& o, const Case& c) { return o << "input: " << c.input; } using ResolverConstEvalArrayInitTest = ResolverTestWithParam; TEST_P(ResolverConstEvalArrayInitTest, Test) { Enable(ast::Extension::kF16); auto& param = GetParam(); auto* expr = param.input.Expr(*this); auto* a = Const("a", expr); WrapInFunction(a); EXPECT_TRUE(r()->Resolve()) << r()->error(); auto* sem = Sem().Get(expr); ASSERT_NE(sem, nullptr); auto* arr = sem->Type()->As(); ASSERT_NE(arr, nullptr); EXPECT_TYPE(sem->ConstantValue()->Type(), sem->Type()); // Constant values should match input values CheckConstant(sem->ConstantValue(), param.input); } template std::vector ArrayInitCases() { return { C(Array(T(0))), // C(Array(T(0))), // C(Array(T(0), T(1))), // C(Array(T(0), T(1), T(2))), // C(Array(T(2), T(1), T(0))), // C(Array(T(2), T(0), T(1))), // }; } INSTANTIATE_TEST_SUITE_P( // ArrayInit, ResolverConstEvalArrayInitTest, testing::ValuesIn(Concat(ArrayInitCases(), // ArrayInitCases(), // ArrayInitCases(), // ArrayInitCases(), // ArrayInitCases(), // ArrayInitCases(), // ArrayInitCases()))); } // namespace ArrayInit TEST_F(ResolverConstEvalTest, ArrayInit_Nested_f32) { auto inner_ty = [&] { return ty.array(); }; auto outer_ty = ty.array(inner_ty(), Expr(3_i)); auto* expr = Construct(outer_ty, // Construct(inner_ty(), 1_f, 2_f), // Construct(inner_ty(), 3_f, 4_f), // Construct(inner_ty(), 5_f, 6_f)); WrapInFunction(expr); EXPECT_TRUE(r()->Resolve()) << r()->error(); auto* sem = Sem().Get(expr); ASSERT_NE(sem, nullptr); auto* outer_arr = sem->Type()->As(); ASSERT_NE(outer_arr, nullptr); EXPECT_TRUE(outer_arr->ElemType()->Is()); EXPECT_TRUE(outer_arr->ElemType()->As()->ElemType()->Is()); auto* arr = sem->ConstantValue(); EXPECT_FALSE(arr->AllEqual()); EXPECT_FALSE(arr->AnyZero()); EXPECT_FALSE(arr->AllZero()); EXPECT_FALSE(arr->Index(0)->AllEqual()); EXPECT_FALSE(arr->Index(0)->AnyZero()); EXPECT_FALSE(arr->Index(0)->AllZero()); EXPECT_FALSE(arr->Index(1)->AllEqual()); EXPECT_FALSE(arr->Index(1)->AnyZero()); EXPECT_FALSE(arr->Index(1)->AllZero()); EXPECT_FALSE(arr->Index(2)->AllEqual()); EXPECT_FALSE(arr->Index(2)->AnyZero()); EXPECT_FALSE(arr->Index(2)->AllZero()); EXPECT_EQ(arr->Index(0)->Index(0)->ValueAs(), 1.0f); EXPECT_EQ(arr->Index(0)->Index(1)->ValueAs(), 2.0f); EXPECT_EQ(arr->Index(1)->Index(0)->ValueAs(), 3.0f); EXPECT_EQ(arr->Index(1)->Index(1)->ValueAs(), 4.0f); EXPECT_EQ(arr->Index(2)->Index(0)->ValueAs(), 5.0f); EXPECT_EQ(arr->Index(2)->Index(1)->ValueAs(), 6.0f); } TEST_F(ResolverConstEvalTest, Array_f32_Elements) { auto* expr = Construct(ty.array(), 10_f, 20_f, 30_f, 40_f); WrapInFunction(expr); EXPECT_TRUE(r()->Resolve()) << r()->error(); auto* sem = Sem().Get(expr); ASSERT_NE(sem, nullptr); auto* arr = sem->Type()->As(); ASSERT_NE(arr, nullptr); EXPECT_TRUE(arr->ElemType()->Is()); EXPECT_TYPE(sem->ConstantValue()->Type(), sem->Type()); EXPECT_FALSE(sem->ConstantValue()->AllEqual()); EXPECT_FALSE(sem->ConstantValue()->AnyZero()); EXPECT_FALSE(sem->ConstantValue()->AllZero()); EXPECT_TRUE(sem->ConstantValue()->Index(0)->AllEqual()); EXPECT_FALSE(sem->ConstantValue()->Index(0)->AnyZero()); EXPECT_FALSE(sem->ConstantValue()->Index(0)->AllZero()); EXPECT_EQ(sem->ConstantValue()->Index(0)->ValueAs(), 10_f); EXPECT_TRUE(sem->ConstantValue()->Index(1)->AllEqual()); EXPECT_FALSE(sem->ConstantValue()->Index(1)->AnyZero()); EXPECT_FALSE(sem->ConstantValue()->Index(1)->AllZero()); EXPECT_EQ(sem->ConstantValue()->Index(1)->ValueAs(), 20_f); EXPECT_TRUE(sem->ConstantValue()->Index(2)->AllEqual()); EXPECT_FALSE(sem->ConstantValue()->Index(2)->AnyZero()); EXPECT_FALSE(sem->ConstantValue()->Index(2)->AllZero()); EXPECT_EQ(sem->ConstantValue()->Index(2)->ValueAs(), 30_f); EXPECT_TRUE(sem->ConstantValue()->Index(3)->AllEqual()); EXPECT_FALSE(sem->ConstantValue()->Index(3)->AnyZero()); EXPECT_FALSE(sem->ConstantValue()->Index(3)->AllZero()); EXPECT_EQ(sem->ConstantValue()->Index(3)->ValueAs(), 40_f); } TEST_F(ResolverConstEvalTest, Array_vec3_f32_Elements) { auto* expr = Construct(ty.array(ty.vec3(), 2_u), // vec3(1_f, 2_f, 3_f), vec3(4_f, 5_f, 6_f)); WrapInFunction(expr); EXPECT_TRUE(r()->Resolve()) << r()->error(); auto* sem = Sem().Get(expr); ASSERT_NE(sem, nullptr); auto* arr = sem->Type()->As(); ASSERT_NE(arr, nullptr); EXPECT_TRUE(arr->ElemType()->Is()); EXPECT_TYPE(sem->ConstantValue()->Type(), sem->Type()); EXPECT_FALSE(sem->ConstantValue()->AllEqual()); EXPECT_FALSE(sem->ConstantValue()->AnyZero()); EXPECT_FALSE(sem->ConstantValue()->AllZero()); EXPECT_EQ(sem->ConstantValue()->Index(0)->Index(0)->ValueAs(), 1_f); EXPECT_EQ(sem->ConstantValue()->Index(0)->Index(1)->ValueAs(), 2_f); EXPECT_EQ(sem->ConstantValue()->Index(0)->Index(2)->ValueAs(), 3_f); EXPECT_EQ(sem->ConstantValue()->Index(1)->Index(0)->ValueAs(), 4_f); EXPECT_EQ(sem->ConstantValue()->Index(1)->Index(1)->ValueAs(), 5_f); EXPECT_EQ(sem->ConstantValue()->Index(1)->Index(2)->ValueAs(), 6_f); } TEST_F(ResolverConstEvalTest, Array_Struct_f32_Elements) { Structure("S", utils::Vector{ Member("m1", ty.f32()), Member("m2", ty.f32()), }); auto* expr = Construct(ty.array(ty.type_name("S"), 2_u), // Construct(ty.type_name("S"), 1_f, 2_f), // Construct(ty.type_name("S"), 3_f, 4_f)); WrapInFunction(expr); EXPECT_TRUE(r()->Resolve()) << r()->error(); auto* sem = Sem().Get(expr); ASSERT_NE(sem, nullptr); auto* arr = sem->Type()->As(); ASSERT_NE(arr, nullptr); EXPECT_TRUE(arr->ElemType()->Is()); EXPECT_TYPE(sem->ConstantValue()->Type(), sem->Type()); EXPECT_FALSE(sem->ConstantValue()->AllEqual()); EXPECT_FALSE(sem->ConstantValue()->AnyZero()); EXPECT_FALSE(sem->ConstantValue()->AllZero()); EXPECT_TRUE(sem->ConstantValue()->Index(0)->Index(0)->AllEqual()); EXPECT_FALSE(sem->ConstantValue()->Index(0)->Index(0)->AnyZero()); EXPECT_FALSE(sem->ConstantValue()->Index(0)->Index(0)->AllZero()); EXPECT_EQ(sem->ConstantValue()->Index(0)->Index(0)->ValueAs(), 1_f); EXPECT_TRUE(sem->ConstantValue()->Index(0)->Index(1)->AllEqual()); EXPECT_FALSE(sem->ConstantValue()->Index(0)->Index(1)->AnyZero()); EXPECT_FALSE(sem->ConstantValue()->Index(0)->Index(1)->AllZero()); EXPECT_EQ(sem->ConstantValue()->Index(0)->Index(1)->ValueAs(), 2_f); EXPECT_TRUE(sem->ConstantValue()->Index(1)->Index(0)->AllEqual()); EXPECT_FALSE(sem->ConstantValue()->Index(1)->Index(0)->AnyZero()); EXPECT_FALSE(sem->ConstantValue()->Index(1)->Index(0)->AllZero()); EXPECT_EQ(sem->ConstantValue()->Index(1)->Index(0)->ValueAs(), 3_f); EXPECT_TRUE(sem->ConstantValue()->Index(1)->Index(1)->AllEqual()); EXPECT_FALSE(sem->ConstantValue()->Index(1)->Index(1)->AnyZero()); EXPECT_FALSE(sem->ConstantValue()->Index(1)->Index(1)->AllZero()); EXPECT_EQ(sem->ConstantValue()->Index(1)->Index(1)->ValueAs(), 4_f); } TEST_F(ResolverConstEvalTest, Struct_ZeroInit) { Enable(ast::Extension::kF16); auto* s = Structure("S", utils::Vector{ Member("a", ty.i32()), Member("b", ty.u32()), Member("c", ty.f32()), Member("d", ty.f16()), Member("e", ty.bool_()), }); auto* expr = Construct(ty.Of(s)); auto* a = Const("a", expr); WrapInFunction(a); EXPECT_TRUE(r()->Resolve()) << r()->error(); auto* sem = Sem().Get(expr); ASSERT_NE(sem, nullptr); auto* str = sem->Type()->As(); ASSERT_NE(str, nullptr); EXPECT_EQ(str->Members().Length(), 5u); EXPECT_TYPE(sem->ConstantValue()->Type(), sem->Type()); EXPECT_FALSE(sem->ConstantValue()->AllEqual()); EXPECT_TRUE(sem->ConstantValue()->AnyZero()); EXPECT_TRUE(sem->ConstantValue()->AllZero()); EXPECT_TRUE(sem->ConstantValue()->Index(0)->Type()->Is()); EXPECT_EQ(sem->ConstantValue()->Index(0)->ValueAs(), 0); EXPECT_TRUE(sem->ConstantValue()->Index(1)->Type()->Is()); EXPECT_EQ(sem->ConstantValue()->Index(1)->ValueAs(), 0u); EXPECT_TRUE(sem->ConstantValue()->Index(2)->Type()->Is()); EXPECT_EQ(sem->ConstantValue()->Index(2)->ValueAs(), 0.0f); EXPECT_TRUE(sem->ConstantValue()->Index(3)->Type()->Is()); EXPECT_EQ(sem->ConstantValue()->Index(3)->ValueAs(), 0.0f); EXPECT_TRUE(sem->ConstantValue()->Index(4)->Type()->Is()); EXPECT_EQ(sem->ConstantValue()->Index(4)->ValueAs(), false); for (size_t i = 0; i < str->Members().Length(); ++i) { EXPECT_TRUE(sem->ConstantValue()->Index(i)->AllEqual()); EXPECT_TRUE(sem->ConstantValue()->Index(i)->AnyZero()); EXPECT_TRUE(sem->ConstantValue()->Index(i)->AllZero()); } } TEST_F(ResolverConstEvalTest, Struct_Nested_ZeroInit) { Enable(ast::Extension::kF16); auto* inner = Structure("Inner", utils::Vector{ Member("a", ty.i32()), Member("b", ty.u32()), Member("c", ty.f32()), Member("d", ty.f16()), Member("e", ty.bool_()), }); auto* s = Structure("s", // utils::Vector{ Member("inner", ty.Of(inner)), }); auto* expr = Construct(ty.Of(s)); auto* a = Const("a", expr); WrapInFunction(a); EXPECT_TRUE(r()->Resolve()) << r()->error(); auto* sem = Sem().Get(expr); ASSERT_NE(sem, nullptr); auto* str = sem->Type()->As(); ASSERT_NE(str, nullptr); EXPECT_EQ(str->Members().Length(), 1u); EXPECT_TYPE(sem->ConstantValue()->Type(), sem->Type()); EXPECT_TRUE(sem->ConstantValue()->AllEqual()); EXPECT_TRUE(sem->ConstantValue()->AnyZero()); EXPECT_TRUE(sem->ConstantValue()->AllZero()); auto* inner_struct = sem->ConstantValue()->Index(0); EXPECT_FALSE(inner_struct->AllEqual()); EXPECT_TRUE(inner_struct->AnyZero()); EXPECT_TRUE(inner_struct->AllZero()); EXPECT_TRUE(inner_struct->Index(0)->Type()->Is()); EXPECT_EQ(inner_struct->Index(0)->ValueAs(), 0); EXPECT_TRUE(inner_struct->Index(1)->Type()->Is()); EXPECT_EQ(inner_struct->Index(1)->ValueAs(), 0u); EXPECT_TRUE(inner_struct->Index(2)->Type()->Is()); EXPECT_EQ(inner_struct->Index(2)->ValueAs(), 0.0f); EXPECT_TRUE(inner_struct->Index(3)->Type()->Is()); EXPECT_EQ(inner_struct->Index(3)->ValueAs(), 0.0f); EXPECT_TRUE(inner_struct->Index(4)->Type()->Is()); EXPECT_EQ(inner_struct->Index(4)->ValueAs(), false); for (size_t i = 0; i < str->Members().Length(); ++i) { EXPECT_TRUE(inner_struct->Index(i)->AllEqual()); EXPECT_TRUE(inner_struct->Index(i)->AnyZero()); EXPECT_TRUE(inner_struct->Index(i)->AllZero()); } } TEST_F(ResolverConstEvalTest, Struct_I32s_ZeroInit) { Structure( "S", utils::Vector{Member("m1", ty.i32()), Member("m2", ty.i32()), Member("m3", ty.i32())}); auto* expr = Construct(ty.type_name("S")); WrapInFunction(expr); EXPECT_TRUE(r()->Resolve()) << r()->error(); auto* sem = Sem().Get(expr); ASSERT_NE(sem, nullptr); auto* str = sem->Type()->As(); ASSERT_NE(str, nullptr); EXPECT_EQ(str->Members().Length(), 3u); ASSERT_NE(sem->ConstantValue(), nullptr); EXPECT_TYPE(sem->ConstantValue()->Type(), sem->Type()); EXPECT_TRUE(sem->ConstantValue()->AllEqual()); EXPECT_TRUE(sem->ConstantValue()->AnyZero()); EXPECT_TRUE(sem->ConstantValue()->AllZero()); EXPECT_TRUE(sem->ConstantValue()->Index(0)->AllEqual()); EXPECT_TRUE(sem->ConstantValue()->Index(0)->AnyZero()); EXPECT_TRUE(sem->ConstantValue()->Index(0)->AllZero()); EXPECT_TRUE(sem->ConstantValue()->Index(0)->Type()->Is()); EXPECT_EQ(sem->ConstantValue()->Index(0)->ValueAs(), 0_i); EXPECT_TRUE(sem->ConstantValue()->Index(1)->AllEqual()); EXPECT_TRUE(sem->ConstantValue()->Index(1)->AnyZero()); EXPECT_TRUE(sem->ConstantValue()->Index(1)->AllZero()); EXPECT_TRUE(sem->ConstantValue()->Index(1)->Type()->Is()); EXPECT_EQ(sem->ConstantValue()->Index(1)->ValueAs(), 0_i); EXPECT_TRUE(sem->ConstantValue()->Index(2)->AllEqual()); EXPECT_TRUE(sem->ConstantValue()->Index(2)->AnyZero()); EXPECT_TRUE(sem->ConstantValue()->Index(2)->AllZero()); EXPECT_TRUE(sem->ConstantValue()->Index(2)->Type()->Is()); EXPECT_EQ(sem->ConstantValue()->Index(2)->ValueAs(), 0_i); } TEST_F(ResolverConstEvalTest, Struct_MixedScalars_ZeroInit) { Enable(ast::Extension::kF16); Structure("S", utils::Vector{ Member("m1", ty.i32()), Member("m2", ty.u32()), Member("m3", ty.f32()), Member("m4", ty.f16()), Member("m5", ty.bool_()), }); auto* expr = Construct(ty.type_name("S")); WrapInFunction(expr); EXPECT_TRUE(r()->Resolve()) << r()->error(); auto* sem = Sem().Get(expr); ASSERT_NE(sem, nullptr); auto* str = sem->Type()->As(); ASSERT_NE(str, nullptr); EXPECT_EQ(str->Members().Length(), 5u); ASSERT_NE(sem->ConstantValue(), nullptr); EXPECT_TYPE(sem->ConstantValue()->Type(), sem->Type()); EXPECT_FALSE(sem->ConstantValue()->AllEqual()); EXPECT_TRUE(sem->ConstantValue()->AnyZero()); EXPECT_TRUE(sem->ConstantValue()->AllZero()); EXPECT_TRUE(sem->ConstantValue()->Index(0)->AllEqual()); EXPECT_TRUE(sem->ConstantValue()->Index(0)->AnyZero()); EXPECT_TRUE(sem->ConstantValue()->Index(0)->AllZero()); EXPECT_TRUE(sem->ConstantValue()->Index(0)->Type()->Is()); EXPECT_EQ(sem->ConstantValue()->Index(0)->ValueAs(), 0_i); EXPECT_TRUE(sem->ConstantValue()->Index(1)->AllEqual()); EXPECT_TRUE(sem->ConstantValue()->Index(1)->AnyZero()); EXPECT_TRUE(sem->ConstantValue()->Index(1)->AllZero()); EXPECT_TRUE(sem->ConstantValue()->Index(1)->Type()->Is()); EXPECT_EQ(sem->ConstantValue()->Index(1)->ValueAs(), 0_u); EXPECT_TRUE(sem->ConstantValue()->Index(2)->AllEqual()); EXPECT_TRUE(sem->ConstantValue()->Index(2)->AnyZero()); EXPECT_TRUE(sem->ConstantValue()->Index(2)->AllZero()); EXPECT_TRUE(sem->ConstantValue()->Index(2)->Type()->Is()); EXPECT_EQ(sem->ConstantValue()->Index(2)->ValueAs(), 0._f); EXPECT_TRUE(sem->ConstantValue()->Index(3)->AllEqual()); EXPECT_TRUE(sem->ConstantValue()->Index(3)->AnyZero()); EXPECT_TRUE(sem->ConstantValue()->Index(3)->AllZero()); EXPECT_TRUE(sem->ConstantValue()->Index(3)->Type()->Is()); EXPECT_EQ(sem->ConstantValue()->Index(3)->ValueAs(), 0._h); EXPECT_TRUE(sem->ConstantValue()->Index(4)->AllEqual()); EXPECT_TRUE(sem->ConstantValue()->Index(4)->AnyZero()); EXPECT_TRUE(sem->ConstantValue()->Index(4)->AllZero()); EXPECT_TRUE(sem->ConstantValue()->Index(4)->Type()->Is()); EXPECT_EQ(sem->ConstantValue()->Index(4)->ValueAs(), false); } TEST_F(ResolverConstEvalTest, Struct_VectorF32s_ZeroInit) { Structure("S", utils::Vector{ Member("m1", ty.vec3()), Member("m2", ty.vec3()), Member("m3", ty.vec3()), }); auto* expr = Construct(ty.type_name("S")); WrapInFunction(expr); EXPECT_TRUE(r()->Resolve()) << r()->error(); auto* sem = Sem().Get(expr); ASSERT_NE(sem, nullptr); auto* str = sem->Type()->As(); ASSERT_NE(str, nullptr); EXPECT_EQ(str->Members().Length(), 3u); ASSERT_NE(sem->ConstantValue(), nullptr); EXPECT_TYPE(sem->ConstantValue()->Type(), sem->Type()); EXPECT_TRUE(sem->ConstantValue()->AllEqual()); EXPECT_TRUE(sem->ConstantValue()->AnyZero()); EXPECT_TRUE(sem->ConstantValue()->AllZero()); EXPECT_TRUE(sem->ConstantValue()->Index(0)->AllEqual()); EXPECT_TRUE(sem->ConstantValue()->Index(0)->AnyZero()); EXPECT_TRUE(sem->ConstantValue()->Index(0)->AllZero()); EXPECT_TRUE(sem->ConstantValue()->Index(0)->Type()->Is()); EXPECT_TRUE( sem->ConstantValue()->Index(0)->Type()->As()->type()->Is()); EXPECT_EQ(sem->ConstantValue()->Index(0)->Index(0)->ValueAs(), 0._f); EXPECT_EQ(sem->ConstantValue()->Index(0)->Index(1)->ValueAs(), 0._f); EXPECT_EQ(sem->ConstantValue()->Index(0)->Index(2)->ValueAs(), 0._f); EXPECT_TRUE(sem->ConstantValue()->Index(1)->AllEqual()); EXPECT_TRUE(sem->ConstantValue()->Index(1)->AnyZero()); EXPECT_TRUE(sem->ConstantValue()->Index(1)->AllZero()); EXPECT_TRUE(sem->ConstantValue()->Index(1)->Type()->Is()); EXPECT_TRUE( sem->ConstantValue()->Index(1)->Type()->As()->type()->Is()); EXPECT_EQ(sem->ConstantValue()->Index(1)->Index(0)->ValueAs(), 0._f); EXPECT_EQ(sem->ConstantValue()->Index(1)->Index(1)->ValueAs(), 0._f); EXPECT_EQ(sem->ConstantValue()->Index(1)->Index(2)->ValueAs(), 0._f); EXPECT_TRUE(sem->ConstantValue()->Index(2)->AllEqual()); EXPECT_TRUE(sem->ConstantValue()->Index(2)->AnyZero()); EXPECT_TRUE(sem->ConstantValue()->Index(2)->AllZero()); EXPECT_TRUE(sem->ConstantValue()->Index(2)->Type()->Is()); EXPECT_TRUE( sem->ConstantValue()->Index(2)->Type()->As()->type()->Is()); EXPECT_EQ(sem->ConstantValue()->Index(2)->Index(0)->ValueAs(), 0._f); EXPECT_EQ(sem->ConstantValue()->Index(2)->Index(1)->ValueAs(), 0._f); EXPECT_EQ(sem->ConstantValue()->Index(2)->Index(2)->ValueAs(), 0._f); } TEST_F(ResolverConstEvalTest, Struct_MixedVectors_ZeroInit) { Enable(ast::Extension::kF16); Structure("S", utils::Vector{ Member("m1", ty.vec2()), Member("m2", ty.vec3()), Member("m3", ty.vec4()), Member("m4", ty.vec3()), Member("m5", ty.vec2()), }); auto* expr = Construct(ty.type_name("S")); WrapInFunction(expr); EXPECT_TRUE(r()->Resolve()) << r()->error(); auto* sem = Sem().Get(expr); ASSERT_NE(sem, nullptr); auto* str = sem->Type()->As(); ASSERT_NE(str, nullptr); EXPECT_EQ(str->Members().Length(), 5u); ASSERT_NE(sem->ConstantValue(), nullptr); EXPECT_TYPE(sem->ConstantValue()->Type(), sem->Type()); EXPECT_FALSE(sem->ConstantValue()->AllEqual()); EXPECT_TRUE(sem->ConstantValue()->AnyZero()); EXPECT_TRUE(sem->ConstantValue()->AllZero()); EXPECT_TRUE(sem->ConstantValue()->Index(0)->AllEqual()); EXPECT_TRUE(sem->ConstantValue()->Index(0)->AnyZero()); EXPECT_TRUE(sem->ConstantValue()->Index(0)->AllZero()); EXPECT_TRUE(sem->ConstantValue()->Index(0)->Type()->Is()); EXPECT_TRUE( sem->ConstantValue()->Index(0)->Type()->As()->type()->Is()); EXPECT_EQ(sem->ConstantValue()->Index(0)->Index(0)->ValueAs(), 0_i); EXPECT_EQ(sem->ConstantValue()->Index(0)->Index(1)->ValueAs(), 0_i); EXPECT_TRUE(sem->ConstantValue()->Index(1)->AllEqual()); EXPECT_TRUE(sem->ConstantValue()->Index(1)->AnyZero()); EXPECT_TRUE(sem->ConstantValue()->Index(1)->AllZero()); EXPECT_TRUE(sem->ConstantValue()->Index(1)->Type()->Is()); EXPECT_TRUE( sem->ConstantValue()->Index(1)->Type()->As()->type()->Is()); EXPECT_EQ(sem->ConstantValue()->Index(1)->Index(0)->ValueAs(), 0_u); EXPECT_EQ(sem->ConstantValue()->Index(1)->Index(1)->ValueAs(), 0_u); EXPECT_EQ(sem->ConstantValue()->Index(1)->Index(2)->ValueAs(), 0_u); EXPECT_TRUE(sem->ConstantValue()->Index(2)->AllEqual()); EXPECT_TRUE(sem->ConstantValue()->Index(2)->AnyZero()); EXPECT_TRUE(sem->ConstantValue()->Index(2)->AllZero()); EXPECT_TRUE(sem->ConstantValue()->Index(2)->Type()->Is()); EXPECT_TRUE( sem->ConstantValue()->Index(2)->Type()->As()->type()->Is()); EXPECT_EQ(sem->ConstantValue()->Index(2)->Index(0)->ValueAs(), 0._f); EXPECT_EQ(sem->ConstantValue()->Index(2)->Index(1)->ValueAs(), 0._f); EXPECT_EQ(sem->ConstantValue()->Index(2)->Index(2)->ValueAs(), 0._f); EXPECT_EQ(sem->ConstantValue()->Index(2)->Index(3)->ValueAs(), 0._f); EXPECT_TRUE(sem->ConstantValue()->Index(3)->AllEqual()); EXPECT_TRUE(sem->ConstantValue()->Index(3)->AnyZero()); EXPECT_TRUE(sem->ConstantValue()->Index(3)->AllZero()); EXPECT_TRUE(sem->ConstantValue()->Index(3)->Type()->Is()); EXPECT_TRUE( sem->ConstantValue()->Index(3)->Type()->As()->type()->Is()); EXPECT_EQ(sem->ConstantValue()->Index(3)->Index(0)->ValueAs(), 0._h); EXPECT_EQ(sem->ConstantValue()->Index(3)->Index(1)->ValueAs(), 0._h); EXPECT_EQ(sem->ConstantValue()->Index(3)->Index(2)->ValueAs(), 0._h); EXPECT_TRUE(sem->ConstantValue()->Index(4)->AllEqual()); EXPECT_TRUE(sem->ConstantValue()->Index(4)->AnyZero()); EXPECT_TRUE(sem->ConstantValue()->Index(4)->AllZero()); EXPECT_TRUE(sem->ConstantValue()->Index(4)->Type()->Is()); EXPECT_TRUE( sem->ConstantValue()->Index(4)->Type()->As()->type()->Is()); EXPECT_EQ(sem->ConstantValue()->Index(4)->Index(0)->ValueAs(), false); EXPECT_EQ(sem->ConstantValue()->Index(4)->Index(1)->ValueAs(), false); } TEST_F(ResolverConstEvalTest, Struct_Struct_ZeroInit) { Structure("Inner", utils::Vector{ Member("m1", ty.i32()), Member("m2", ty.u32()), Member("m3", ty.f32()), }); Structure("Outer", utils::Vector{ Member("m1", ty.type_name("Inner")), Member("m2", ty.type_name("Inner")), }); auto* expr = Construct(ty.type_name("Outer")); WrapInFunction(expr); EXPECT_TRUE(r()->Resolve()) << r()->error(); auto* sem = Sem().Get(expr); ASSERT_NE(sem, nullptr); auto* str = sem->Type()->As(); ASSERT_NE(str, nullptr); EXPECT_EQ(str->Members().Length(), 2u); ASSERT_NE(sem->ConstantValue(), nullptr); EXPECT_TYPE(sem->ConstantValue()->Type(), sem->Type()); EXPECT_TRUE(sem->ConstantValue()->AllEqual()); EXPECT_TRUE(sem->ConstantValue()->AnyZero()); EXPECT_TRUE(sem->ConstantValue()->AllZero()); EXPECT_FALSE(sem->ConstantValue()->Index(0)->AllEqual()); EXPECT_TRUE(sem->ConstantValue()->Index(0)->AnyZero()); EXPECT_TRUE(sem->ConstantValue()->Index(0)->AllZero()); EXPECT_TRUE(sem->ConstantValue()->Index(0)->Type()->Is()); EXPECT_EQ(sem->ConstantValue()->Index(0)->Index(0)->ValueAs(), 0_i); EXPECT_EQ(sem->ConstantValue()->Index(0)->Index(1)->ValueAs(), 0_u); EXPECT_EQ(sem->ConstantValue()->Index(0)->Index(2)->ValueAs(), 0_f); EXPECT_FALSE(sem->ConstantValue()->Index(1)->AllEqual()); EXPECT_TRUE(sem->ConstantValue()->Index(1)->AnyZero()); EXPECT_TRUE(sem->ConstantValue()->Index(1)->AllZero()); EXPECT_TRUE(sem->ConstantValue()->Index(1)->Type()->Is()); EXPECT_EQ(sem->ConstantValue()->Index(1)->Index(0)->ValueAs(), 0_i); EXPECT_EQ(sem->ConstantValue()->Index(1)->Index(1)->ValueAs(), 0_u); EXPECT_EQ(sem->ConstantValue()->Index(1)->Index(2)->ValueAs(), 0_f); } TEST_F(ResolverConstEvalTest, Struct_MixedScalars_Construct) { Enable(ast::Extension::kF16); Structure("S", utils::Vector{ Member("m1", ty.i32()), Member("m2", ty.u32()), Member("m3", ty.f32()), Member("m4", ty.f16()), Member("m5", ty.bool_()), }); auto* expr = Construct(ty.type_name("S"), 1_i, 2_u, 3_f, 4_h, false); WrapInFunction(expr); EXPECT_TRUE(r()->Resolve()) << r()->error(); auto* sem = Sem().Get(expr); ASSERT_NE(sem, nullptr); auto* str = sem->Type()->As(); ASSERT_NE(str, nullptr); EXPECT_EQ(str->Members().Length(), 5u); ASSERT_NE(sem->ConstantValue(), nullptr); EXPECT_TYPE(sem->ConstantValue()->Type(), sem->Type()); EXPECT_FALSE(sem->ConstantValue()->AllEqual()); EXPECT_TRUE(sem->ConstantValue()->AnyZero()); EXPECT_FALSE(sem->ConstantValue()->AllZero()); EXPECT_TRUE(sem->ConstantValue()->Index(0)->AllEqual()); EXPECT_FALSE(sem->ConstantValue()->Index(0)->AnyZero()); EXPECT_FALSE(sem->ConstantValue()->Index(0)->AllZero()); EXPECT_TRUE(sem->ConstantValue()->Index(0)->Type()->Is()); EXPECT_EQ(sem->ConstantValue()->Index(0)->ValueAs(), 1_i); EXPECT_TRUE(sem->ConstantValue()->Index(1)->AllEqual()); EXPECT_FALSE(sem->ConstantValue()->Index(1)->AnyZero()); EXPECT_FALSE(sem->ConstantValue()->Index(1)->AllZero()); EXPECT_TRUE(sem->ConstantValue()->Index(1)->Type()->Is()); EXPECT_EQ(sem->ConstantValue()->Index(1)->ValueAs(), 2_u); EXPECT_TRUE(sem->ConstantValue()->Index(2)->AllEqual()); EXPECT_FALSE(sem->ConstantValue()->Index(2)->AnyZero()); EXPECT_FALSE(sem->ConstantValue()->Index(2)->AllZero()); EXPECT_TRUE(sem->ConstantValue()->Index(2)->Type()->Is()); EXPECT_EQ(sem->ConstantValue()->Index(2)->ValueAs(), 3._f); EXPECT_TRUE(sem->ConstantValue()->Index(3)->AllEqual()); EXPECT_FALSE(sem->ConstantValue()->Index(3)->AnyZero()); EXPECT_FALSE(sem->ConstantValue()->Index(3)->AllZero()); EXPECT_TRUE(sem->ConstantValue()->Index(3)->Type()->Is()); EXPECT_EQ(sem->ConstantValue()->Index(3)->ValueAs(), 4._h); EXPECT_TRUE(sem->ConstantValue()->Index(4)->AllEqual()); EXPECT_TRUE(sem->ConstantValue()->Index(4)->AnyZero()); EXPECT_TRUE(sem->ConstantValue()->Index(4)->AllZero()); EXPECT_TRUE(sem->ConstantValue()->Index(4)->Type()->Is()); EXPECT_EQ(sem->ConstantValue()->Index(4)->ValueAs(), false); } TEST_F(ResolverConstEvalTest, Struct_MixedVectors_Construct) { Enable(ast::Extension::kF16); Structure("S", utils::Vector{ Member("m1", ty.vec2()), Member("m2", ty.vec3()), Member("m3", ty.vec4()), Member("m4", ty.vec3()), Member("m5", ty.vec2()), }); auto* expr = Construct(ty.type_name("S"), vec2(1_i), vec3(2_u), vec4(3_f), vec3(4_h), vec2(false)); WrapInFunction(expr); EXPECT_TRUE(r()->Resolve()) << r()->error(); auto* sem = Sem().Get(expr); ASSERT_NE(sem, nullptr); auto* str = sem->Type()->As(); ASSERT_NE(str, nullptr); EXPECT_EQ(str->Members().Length(), 5u); ASSERT_NE(sem->ConstantValue(), nullptr); EXPECT_TYPE(sem->ConstantValue()->Type(), sem->Type()); EXPECT_FALSE(sem->ConstantValue()->AllEqual()); EXPECT_TRUE(sem->ConstantValue()->AnyZero()); EXPECT_FALSE(sem->ConstantValue()->AllZero()); EXPECT_TRUE(sem->ConstantValue()->Index(0)->AllEqual()); EXPECT_FALSE(sem->ConstantValue()->Index(0)->AnyZero()); EXPECT_FALSE(sem->ConstantValue()->Index(0)->AllZero()); EXPECT_TRUE(sem->ConstantValue()->Index(0)->Type()->Is()); EXPECT_TRUE( sem->ConstantValue()->Index(0)->Type()->As()->type()->Is()); EXPECT_EQ(sem->ConstantValue()->Index(0)->Index(0)->ValueAs(), 1_i); EXPECT_EQ(sem->ConstantValue()->Index(0)->Index(1)->ValueAs(), 1_i); EXPECT_TRUE(sem->ConstantValue()->Index(1)->AllEqual()); EXPECT_FALSE(sem->ConstantValue()->Index(1)->AnyZero()); EXPECT_FALSE(sem->ConstantValue()->Index(1)->AllZero()); EXPECT_TRUE(sem->ConstantValue()->Index(1)->Type()->Is()); EXPECT_TRUE( sem->ConstantValue()->Index(1)->Type()->As()->type()->Is()); EXPECT_EQ(sem->ConstantValue()->Index(1)->Index(0)->ValueAs(), 2_u); EXPECT_EQ(sem->ConstantValue()->Index(1)->Index(1)->ValueAs(), 2_u); EXPECT_EQ(sem->ConstantValue()->Index(1)->Index(2)->ValueAs(), 2_u); EXPECT_TRUE(sem->ConstantValue()->Index(2)->AllEqual()); EXPECT_FALSE(sem->ConstantValue()->Index(2)->AnyZero()); EXPECT_FALSE(sem->ConstantValue()->Index(2)->AllZero()); EXPECT_TRUE(sem->ConstantValue()->Index(2)->Type()->Is()); EXPECT_TRUE( sem->ConstantValue()->Index(2)->Type()->As()->type()->Is()); EXPECT_EQ(sem->ConstantValue()->Index(2)->Index(0)->ValueAs(), 3._f); EXPECT_EQ(sem->ConstantValue()->Index(2)->Index(1)->ValueAs(), 3._f); EXPECT_EQ(sem->ConstantValue()->Index(2)->Index(2)->ValueAs(), 3._f); EXPECT_EQ(sem->ConstantValue()->Index(2)->Index(3)->ValueAs(), 3._f); EXPECT_TRUE(sem->ConstantValue()->Index(3)->AllEqual()); EXPECT_FALSE(sem->ConstantValue()->Index(3)->AnyZero()); EXPECT_FALSE(sem->ConstantValue()->Index(3)->AllZero()); EXPECT_TRUE(sem->ConstantValue()->Index(3)->Type()->Is()); EXPECT_TRUE( sem->ConstantValue()->Index(3)->Type()->As()->type()->Is()); EXPECT_EQ(sem->ConstantValue()->Index(3)->Index(0)->ValueAs(), 4._h); EXPECT_EQ(sem->ConstantValue()->Index(3)->Index(1)->ValueAs(), 4._h); EXPECT_EQ(sem->ConstantValue()->Index(3)->Index(2)->ValueAs(), 4._h); EXPECT_TRUE(sem->ConstantValue()->Index(4)->AllEqual()); EXPECT_TRUE(sem->ConstantValue()->Index(4)->AnyZero()); EXPECT_TRUE(sem->ConstantValue()->Index(4)->AllZero()); EXPECT_TRUE(sem->ConstantValue()->Index(4)->Type()->Is()); EXPECT_TRUE( sem->ConstantValue()->Index(4)->Type()->As()->type()->Is()); EXPECT_EQ(sem->ConstantValue()->Index(4)->Index(0)->ValueAs(), false); EXPECT_EQ(sem->ConstantValue()->Index(4)->Index(1)->ValueAs(), false); } TEST_F(ResolverConstEvalTest, Struct_Struct_Construct) { Structure("Inner", utils::Vector{ Member("m1", ty.i32()), Member("m2", ty.u32()), Member("m3", ty.f32()), }); Structure("Outer", utils::Vector{ Member("m1", ty.type_name("Inner")), Member("m2", ty.type_name("Inner")), }); auto* expr = Construct(ty.type_name("Outer"), // Construct(ty.type_name("Inner"), 1_i, 2_u, 3_f), Construct(ty.type_name("Inner"), 4_i, 0_u, 6_f)); WrapInFunction(expr); EXPECT_TRUE(r()->Resolve()) << r()->error(); auto* sem = Sem().Get(expr); ASSERT_NE(sem, nullptr); auto* str = sem->Type()->As(); ASSERT_NE(str, nullptr); EXPECT_EQ(str->Members().Length(), 2u); ASSERT_NE(sem->ConstantValue(), nullptr); EXPECT_TYPE(sem->ConstantValue()->Type(), sem->Type()); EXPECT_FALSE(sem->ConstantValue()->AllEqual()); EXPECT_TRUE(sem->ConstantValue()->AnyZero()); EXPECT_FALSE(sem->ConstantValue()->AllZero()); EXPECT_FALSE(sem->ConstantValue()->Index(0)->AllEqual()); EXPECT_FALSE(sem->ConstantValue()->Index(0)->AnyZero()); EXPECT_FALSE(sem->ConstantValue()->Index(0)->AllZero()); EXPECT_TRUE(sem->ConstantValue()->Index(0)->Type()->Is()); EXPECT_EQ(sem->ConstantValue()->Index(0)->Index(0)->ValueAs(), 1_i); EXPECT_EQ(sem->ConstantValue()->Index(0)->Index(1)->ValueAs(), 2_u); EXPECT_EQ(sem->ConstantValue()->Index(0)->Index(2)->ValueAs(), 3_f); EXPECT_FALSE(sem->ConstantValue()->Index(1)->AllEqual()); EXPECT_TRUE(sem->ConstantValue()->Index(1)->AnyZero()); EXPECT_FALSE(sem->ConstantValue()->Index(1)->AllZero()); EXPECT_TRUE(sem->ConstantValue()->Index(1)->Type()->Is()); EXPECT_EQ(sem->ConstantValue()->Index(1)->Index(0)->ValueAs(), 4_i); EXPECT_EQ(sem->ConstantValue()->Index(1)->Index(1)->ValueAs(), 0_u); EXPECT_EQ(sem->ConstantValue()->Index(1)->Index(2)->ValueAs(), 6_f); } TEST_F(ResolverConstEvalTest, Struct_Array_Construct) { Structure("S", utils::Vector{ Member("m1", ty.array()), Member("m2", ty.array()), }); auto* expr = Construct(ty.type_name("S"), // Construct(ty.array(), 1_i, 2_i), Construct(ty.array(), 1_f, 2_f, 3_f)); WrapInFunction(expr); EXPECT_TRUE(r()->Resolve()) << r()->error(); auto* sem = Sem().Get(expr); ASSERT_NE(sem, nullptr); auto* str = sem->Type()->As(); ASSERT_NE(str, nullptr); EXPECT_EQ(str->Members().Length(), 2u); ASSERT_NE(sem->ConstantValue(), nullptr); EXPECT_TYPE(sem->ConstantValue()->Type(), sem->Type()); EXPECT_FALSE(sem->ConstantValue()->AllEqual()); EXPECT_FALSE(sem->ConstantValue()->AnyZero()); EXPECT_FALSE(sem->ConstantValue()->AllZero()); EXPECT_FALSE(sem->ConstantValue()->Index(0)->AllEqual()); EXPECT_FALSE(sem->ConstantValue()->Index(0)->AnyZero()); EXPECT_FALSE(sem->ConstantValue()->Index(0)->AllZero()); EXPECT_TRUE(sem->ConstantValue()->Index(0)->Type()->Is()); EXPECT_EQ(sem->ConstantValue()->Index(0)->Index(0)->ValueAs(), 1_i); EXPECT_EQ(sem->ConstantValue()->Index(0)->Index(1)->ValueAs(), 2_i); EXPECT_FALSE(sem->ConstantValue()->Index(1)->AllEqual()); EXPECT_FALSE(sem->ConstantValue()->Index(1)->AnyZero()); EXPECT_FALSE(sem->ConstantValue()->Index(1)->AllZero()); EXPECT_TRUE(sem->ConstantValue()->Index(1)->Type()->Is()); EXPECT_EQ(sem->ConstantValue()->Index(1)->Index(0)->ValueAs(), 1_f); EXPECT_EQ(sem->ConstantValue()->Index(1)->Index(1)->ValueAs(), 2_f); EXPECT_EQ(sem->ConstantValue()->Index(1)->Index(2)->ValueAs(), 3_f); } } // namespace } // namespace tint::resolver