src/tint/lang/wgsl/resolver/evaluation_stage_test.cc - tint - Git at Google

 // Copyright 2022 The Dawn & Tint Authors
 //
 // Redistribution and use in source and binary forms, with or without
 // modification, are permitted provided that the following conditions are met:
 //
 // 1. Redistributions of source code must retain the above copyright notice, this
 //    list of conditions and the following disclaimer.
 //
 // 2. Redistributions in binary form must reproduce the above copyright notice,
 //    this list of conditions and the following disclaimer in the documentation
 //    and/or other materials provided with the distribution.
 //
 // 3. Neither the name of the copyright holder nor the names of its
 //    contributors may be used to endorse or promote products derived from
 //    this software without specific prior written permission.
 //
 // THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
 // AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
 // IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
 // DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS BE LIABLE
 // FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
 // DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR
 // SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER
 // CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY,
 // OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
 // OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.

 #include "src/tint/lang/wgsl/resolver/resolver.h"

 #include "gmock/gmock.h"
 #include "src/tint/lang/wgsl/resolver/resolver_helper_test.h"
 #include "src/tint/utils/containers/slice.h"

 namespace tint::resolver {
 namespace {

 using namespace tint::core::fluent_types;     // NOLINT
 using namespace tint::core::number_suffixes;  // NOLINT

 using ResolverEvaluationStageTest = ResolverTest;

 TEST_F(ResolverEvaluationStageTest, Literal_i32) {
     auto* expr = Expr(123_i);
     WrapInFunction(expr);

     ASSERT_TRUE(r()->Resolve()) << r()->error();
     EXPECT_EQ(Sem().Get(expr)->Stage(), core::EvaluationStage::kConstant);
 }

 TEST_F(ResolverEvaluationStageTest, Literal_f32) {
     auto* expr = Expr(123_f);
     WrapInFunction(expr);

     ASSERT_TRUE(r()->Resolve()) << r()->error();
     EXPECT_EQ(Sem().Get(expr)->Stage(), core::EvaluationStage::kConstant);
 }

 TEST_F(ResolverEvaluationStageTest, Vector_Init) {
     auto* expr = Call<vec3<f32>>();
     WrapInFunction(expr);

     ASSERT_TRUE(r()->Resolve()) << r()->error();
     EXPECT_EQ(Sem().Get(expr)->Stage(), core::EvaluationStage::kConstant);
 }

 TEST_F(ResolverEvaluationStageTest, Vector_Init_Const_Const) {
     // const f = 1.f;
     // vec2<f32>(f, f);
     auto* f = Const("f", Expr(1_f));
     auto* expr = Call<vec2<f32>>(f, f);
     WrapInFunction(f, expr);

     ASSERT_TRUE(r()->Resolve()) << r()->error();
     EXPECT_EQ(Sem().Get(f)->Stage(), core::EvaluationStage::kConstant);
     EXPECT_EQ(Sem().Get(expr)->Stage(), core::EvaluationStage::kConstant);
 }

 TEST_F(ResolverEvaluationStageTest, Vector_Init_Runtime_Runtime) {
     // var f = 1.f;
     // vec2<f32>(f, f);
     auto* f = Var("f", Expr(1_f));
     auto* expr = Call<vec2<f32>>(f, f);
     WrapInFunction(f, expr);

     ASSERT_TRUE(r()->Resolve()) << r()->error();
     EXPECT_EQ(Sem().Get(f)->Stage(), core::EvaluationStage::kRuntime);
     EXPECT_EQ(Sem().Get(expr)->Stage(), core::EvaluationStage::kRuntime);
 }

 TEST_F(ResolverEvaluationStageTest, Vector_Conv_Const) {
     // const f = 1.f;
     // vec2<u32>(vec2<f32>(f));
     auto* f = Const("f", Expr(1_f));
     auto* expr = Call<vec2<u32>>(Call<vec2<f32>>(f));
     WrapInFunction(f, expr);

     ASSERT_TRUE(r()->Resolve()) << r()->error();
     EXPECT_EQ(Sem().Get(f)->Stage(), core::EvaluationStage::kConstant);
     EXPECT_EQ(Sem().Get(expr)->Stage(), core::EvaluationStage::kConstant);
 }

 TEST_F(ResolverEvaluationStageTest, Vector_Conv_Runtime) {
     // var f = 1.f;
     // vec2<u32>(vec2<f32>(f));
     auto* f = Var("f", Expr(1_f));
     auto* expr = Call<vec2<u32>>(Call<vec2<f32>>(f));
     WrapInFunction(f, expr);

     ASSERT_TRUE(r()->Resolve()) << r()->error();
     EXPECT_EQ(Sem().Get(f)->Stage(), core::EvaluationStage::kRuntime);
     EXPECT_EQ(Sem().Get(expr)->Stage(), core::EvaluationStage::kRuntime);
 }

 TEST_F(ResolverEvaluationStageTest, Matrix_Init) {
     auto* expr = Call<mat2x2<f32>>();
     WrapInFunction(expr);

     ASSERT_TRUE(r()->Resolve()) << r()->error();
     EXPECT_EQ(Sem().Get(expr)->Stage(), core::EvaluationStage::kConstant);
 }

 TEST_F(ResolverEvaluationStageTest, Array_Init) {
     auto* expr = Call<array<f32, 3>>();
     WrapInFunction(expr);

     ASSERT_TRUE(r()->Resolve()) << r()->error();
     EXPECT_EQ(Sem().Get(expr)->Stage(), core::EvaluationStage::kConstant);
 }

 TEST_F(ResolverEvaluationStageTest, Array_Init_Const_Const) {
     // const f = 1.f;
     // array<f32, 2>(f, f);
     auto* f = Const("f", Expr(1_f));
     auto* expr = Call<array<f32, 2>>(f, f);
     WrapInFunction(f, expr);

     ASSERT_TRUE(r()->Resolve()) << r()->error();
     EXPECT_EQ(Sem().Get(f)->Stage(), core::EvaluationStage::kConstant);
     EXPECT_EQ(Sem().Get(expr)->Stage(), core::EvaluationStage::kConstant);
 }

 TEST_F(ResolverEvaluationStageTest, Array_Init_Const_Override) {
     // const f1 = 1.f;
     // override f2 = 2.f;
     // array<f32, 2>(f1, f2);
     auto* f1 = Const("f1", Expr(1_f));
     auto* f2 = Override("f2", Expr(2_f));
     auto* expr = Call<array<f32, 2>>(f1, f2);
     WrapInFunction(f1, expr);

     ASSERT_TRUE(r()->Resolve()) << r()->error();
     EXPECT_EQ(Sem().Get(f1)->Stage(), core::EvaluationStage::kConstant);
     EXPECT_EQ(Sem().Get(f2)->Stage(), core::EvaluationStage::kOverride);
     EXPECT_EQ(Sem().Get(expr)->Stage(), core::EvaluationStage::kOverride);
 }

 TEST_F(ResolverEvaluationStageTest, Array_Init_Override_Runtime) {
     // override f1 = 1.f;
     // var f2 = 2.f;
     // array<f32, 2>(f1, f2);
     auto* f1 = Override("f1", Expr(1_f));
     auto* f2 = Var("f2", Expr(2_f));
     auto* expr = Call<array<f32, 2>>(f1, f2);
     WrapInFunction(f2, expr);

     ASSERT_TRUE(r()->Resolve()) << r()->error();
     EXPECT_EQ(Sem().Get(f1)->Stage(), core::EvaluationStage::kOverride);
     EXPECT_EQ(Sem().Get(f2)->Stage(), core::EvaluationStage::kRuntime);
     EXPECT_EQ(Sem().Get(expr)->Stage(), core::EvaluationStage::kRuntime);
 }

 TEST_F(ResolverEvaluationStageTest, Array_Init_Const_Runtime) {
     // const f1 = 1.f;
     // var f2 = 2.f;
     // array<f32, 2>(f1, f2);
     auto* f1 = Const("f1", Expr(1_f));
     auto* f2 = Var("f2", Expr(2_f));
     auto* expr = Call<array<f32, 2>>(f1, f2);
     WrapInFunction(f1, f2, expr);

     ASSERT_TRUE(r()->Resolve()) << r()->error();
     EXPECT_EQ(Sem().Get(f1)->Stage(), core::EvaluationStage::kConstant);
     EXPECT_EQ(Sem().Get(f2)->Stage(), core::EvaluationStage::kRuntime);
     EXPECT_EQ(Sem().Get(expr)->Stage(), core::EvaluationStage::kRuntime);
 }

 TEST_F(ResolverEvaluationStageTest, Array_Init_Runtime_Runtime) {
     // var f = 1.f;
     // array<f32, 2>(f, f);
     auto* f = Var("f", Expr(1_f));
     auto* expr = Call<array<f32, 2>>(f, f);
     WrapInFunction(f, expr);

     ASSERT_TRUE(r()->Resolve()) << r()->error();
     EXPECT_EQ(Sem().Get(f)->Stage(), core::EvaluationStage::kRuntime);
     EXPECT_EQ(Sem().Get(expr)->Stage(), core::EvaluationStage::kRuntime);
 }

 TEST_F(ResolverEvaluationStageTest, IndexAccessor_Const_Const) {
     // const vec = vec4<f32>();
     // const idx = 1_i;
     // vec[idx]
     auto* vec = Const("vec", Call<vec4<f32>>());
     auto* idx = Const("idx", Expr(1_i));
     auto* expr = IndexAccessor(vec, idx);
     WrapInFunction(vec, idx, expr);

     ASSERT_TRUE(r()->Resolve()) << r()->error();
     EXPECT_EQ(Sem().Get(vec)->Stage(), core::EvaluationStage::kConstant);
     EXPECT_EQ(Sem().Get(idx)->Stage(), core::EvaluationStage::kConstant);
     EXPECT_EQ(Sem().Get(expr)->Stage(), core::EvaluationStage::kConstant);
 }

 TEST_F(ResolverEvaluationStageTest, IndexAccessor_Runtime_Const) {
     // var vec = vec4<f32>();
     // const idx = 1_i;
     // vec[idx]
     auto* vec = Var("vec", Call<vec4<f32>>());
     auto* idx = Const("idx", Expr(1_i));
     auto* expr = IndexAccessor(vec, idx);
     WrapInFunction(vec, idx, expr);

     ASSERT_TRUE(r()->Resolve()) << r()->error();
     EXPECT_EQ(Sem().Get(vec)->Stage(), core::EvaluationStage::kRuntime);
     EXPECT_EQ(Sem().Get(idx)->Stage(), core::EvaluationStage::kConstant);
     EXPECT_EQ(Sem().Get(expr)->Stage(), core::EvaluationStage::kRuntime);
 }

 TEST_F(ResolverEvaluationStageTest, IndexAccessor_Const_Override) {
     // const vec = vec4<f32>();
     // override idx = 1_i;
     // vec[idx]
     auto* vec = Const("vec", Call<vec4<f32>>());
     auto* idx = Override("idx", Expr(1_i));
     auto* expr = IndexAccessor(vec, idx);
     WrapInFunction(vec, expr);

     ASSERT_TRUE(r()->Resolve()) << r()->error();
     EXPECT_EQ(Sem().Get(vec)->Stage(), core::EvaluationStage::kConstant);
     EXPECT_EQ(Sem().Get(idx)->Stage(), core::EvaluationStage::kOverride);
     EXPECT_EQ(Sem().Get(expr)->Stage(), core::EvaluationStage::kOverride);
 }

 TEST_F(ResolverEvaluationStageTest, IndexAccessor_Const_Runtime) {
     // const vec = vec4<f32>();
     // let idx = 1_i;
     // vec[idx]
     auto* vec = Const("vec", Call<vec4<f32>>());
     auto* idx = Let("idx", Expr(1_i));
     auto* expr = IndexAccessor(vec, idx);
     WrapInFunction(vec, idx, expr);

     ASSERT_TRUE(r()->Resolve()) << r()->error();
     EXPECT_EQ(Sem().Get(vec)->Stage(), core::EvaluationStage::kConstant);
     EXPECT_EQ(Sem().Get(idx)->Stage(), core::EvaluationStage::kRuntime);
     EXPECT_EQ(Sem().Get(expr)->Stage(), core::EvaluationStage::kRuntime);
 }

 TEST_F(ResolverEvaluationStageTest, Swizzle_Const) {
     // const vec = S();
     // vec.m
     auto* vec = Const("vec", Call<vec4<f32>>());
     auto* expr = MemberAccessor(vec, "xz");
     WrapInFunction(vec, expr);

     ASSERT_TRUE(r()->Resolve()) << r()->error();
     EXPECT_EQ(Sem().Get(vec)->Stage(), core::EvaluationStage::kConstant);
     EXPECT_EQ(Sem().Get(expr)->Stage(), core::EvaluationStage::kConstant);
 }

 TEST_F(ResolverEvaluationStageTest, Swizzle_Runtime) {
     // var vec = S();
     // vec.m
     auto* vec = Var("vec", Call<vec4<f32>>());
     auto* expr = MemberAccessor(vec, "rg");
     WrapInFunction(vec, expr);

     ASSERT_TRUE(r()->Resolve()) << r()->error();
     EXPECT_EQ(Sem().Get(vec)->Stage(), core::EvaluationStage::kRuntime);
     EXPECT_EQ(Sem().Get(expr)->Stage(), core::EvaluationStage::kRuntime);
 }

 TEST_F(ResolverEvaluationStageTest, MemberAccessor_Const) {
     // struct S { m : i32 };
     // const str = S();
     // str.m
     Structure("S", Vector{Member("m", ty.i32())});
     auto* str = Const("str", Call("S"));
     auto* expr = MemberAccessor(str, "m");
     WrapInFunction(str, expr);

     ASSERT_TRUE(r()->Resolve()) << r()->error();
     EXPECT_EQ(Sem().Get(str)->Stage(), core::EvaluationStage::kConstant);
     EXPECT_EQ(Sem().Get(expr)->Stage(), core::EvaluationStage::kConstant);
 }

 TEST_F(ResolverEvaluationStageTest, MemberAccessor_Runtime) {
     // struct S { m : i32 };
     // var str = S();
     // str.m
     Structure("S", Vector{Member("m", ty.i32())});
     auto* str = Var("str", Call("S"));
     auto* expr = MemberAccessor(str, "m");
     WrapInFunction(str, expr);

     ASSERT_TRUE(r()->Resolve()) << r()->error();
     EXPECT_EQ(Sem().Get(str)->Stage(), core::EvaluationStage::kRuntime);
     EXPECT_EQ(Sem().Get(expr)->Stage(), core::EvaluationStage::kRuntime);
 }

 TEST_F(ResolverEvaluationStageTest, Binary_Runtime) {
     // let one = 1;
     // let result = (one == 1) && (one == 1);
     auto* one = Let("one", Expr(1_a));
     auto* lhs = Equal("one", 1_a);
     auto* rhs = Equal("one", 1_a);
     auto* binary = LogicalAnd(lhs, rhs);
     auto* result = Let("result", binary);
     WrapInFunction(one, result);

     ASSERT_TRUE(r()->Resolve()) << r()->error();
     EXPECT_EQ(Sem().Get(lhs)->Stage(), core::EvaluationStage::kRuntime);
     EXPECT_EQ(Sem().Get(rhs)->Stage(), core::EvaluationStage::kRuntime);
     EXPECT_EQ(Sem().Get(binary)->Stage(), core::EvaluationStage::kRuntime);
 }

 TEST_F(ResolverEvaluationStageTest, Binary_Const) {
     // const one = 1;
     // const result = (one == 1) && (one == 1);
     auto* one = Const("one", Expr(1_a));
     auto* lhs = Equal("one", 1_a);
     auto* rhs = Equal("one", 1_a);
     auto* binary = LogicalAnd(lhs, rhs);
     auto* result = Const("result", binary);
     WrapInFunction(one, result);

     ASSERT_TRUE(r()->Resolve()) << r()->error();
     EXPECT_EQ(Sem().Get(lhs)->Stage(), core::EvaluationStage::kConstant);
     EXPECT_EQ(Sem().Get(rhs)->Stage(), core::EvaluationStage::kConstant);
     EXPECT_EQ(Sem().Get(binary)->Stage(), core::EvaluationStage::kConstant);
 }

 TEST_F(ResolverEvaluationStageTest, Binary_NotEvaluated) {
     // const one = 1;
     // const result = (one == 0) && (one == 1);
     auto* one = Const("one", Expr(1_a));
     auto* lhs = Equal("one", 0_a);
     auto* rhs = Equal("one", 1_a);
     auto* binary = LogicalAnd(lhs, rhs);
     auto* result = Const("result", binary);
     WrapInFunction(one, result);

     ASSERT_TRUE(r()->Resolve()) << r()->error();
     EXPECT_EQ(Sem().Get(lhs)->Stage(), core::EvaluationStage::kConstant);
     EXPECT_EQ(Sem().Get(rhs)->Stage(), core::EvaluationStage::kNotEvaluated);
     EXPECT_EQ(Sem().Get(binary)->Stage(), core::EvaluationStage::kConstant);
 }

 TEST_F(ResolverEvaluationStageTest, FnCall_Runtime) {
     // fn f() -> bool { return true; }
     // let l = false
     // let result = l && f();
     Func("f", Empty, ty.bool_(), Vector{Return(true)});
     auto* let = Let("l", Expr(false));
     auto* lhs = Expr(let);
     auto* rhs = Call("f");
     auto* binary = LogicalAnd(lhs, rhs);
     auto* result = Let("result", binary);
     WrapInFunction(let, result);

     ASSERT_TRUE(r()->Resolve()) << r()->error();
     EXPECT_EQ(Sem().Get(rhs)->Stage(), core::EvaluationStage::kRuntime);
     EXPECT_EQ(Sem().Get(binary)->Stage(), core::EvaluationStage::kRuntime);
 }

 TEST_F(ResolverEvaluationStageTest, FnCall_NotEvaluated) {
     // fn f() -> bool { return true; }
     // let result = false && f();
     Func("f", Empty, ty.bool_(), Vector{Return(true)});
     auto* rhs = Call("f");
     auto* lhs = Expr(false);
     auto* binary = LogicalAnd(lhs, rhs);
     auto* result = Let("result", binary);
     WrapInFunction(result);

     ASSERT_TRUE(r()->Resolve()) << r()->error();
     EXPECT_EQ(Sem().Get(rhs)->Stage(), core::EvaluationStage::kNotEvaluated);
     EXPECT_EQ(Sem().Get(binary)->Stage(), core::EvaluationStage::kConstant);
 }

 TEST_F(ResolverEvaluationStageTest, NestedFnCall_NotEvaluated) {
     // fn f(b : bool) -> bool { return b; }
     // let result = false && f(f(f(1 == 0)));
     Func("f", Vector{Param("b", ty.bool_())}, ty.bool_(), Vector{Return("b")});
     auto* cmp = Equal(0_i, 1_i);
     auto* rhs_0 = Call("f", cmp);
     auto* rhs_1 = Call("f", rhs_0);
     auto* rhs_2 = Call("f", rhs_1);
     auto* lhs = Expr(false);
     auto* binary = LogicalAnd(lhs, rhs_2);
     auto* result = Let("result", binary);
     WrapInFunction(result);

     ASSERT_TRUE(r()->Resolve()) << r()->error();
     EXPECT_EQ(Sem().Get(cmp)->Stage(), core::EvaluationStage::kNotEvaluated);
     EXPECT_EQ(Sem().Get(rhs_0)->Stage(), core::EvaluationStage::kNotEvaluated);
     EXPECT_EQ(Sem().Get(rhs_1)->Stage(), core::EvaluationStage::kNotEvaluated);
     EXPECT_EQ(Sem().Get(rhs_2)->Stage(), core::EvaluationStage::kNotEvaluated);
     EXPECT_EQ(Sem().Get(binary)->Stage(), core::EvaluationStage::kConstant);
 }

 }  // namespace
 }  // namespace tint::resolver
	// Copyright 2022 The Dawn & Tint Authors
	//
	// Redistribution and use in source and binary forms, with or without
	// modification, are permitted provided that the following conditions are met:
	//
	// 1. Redistributions of source code must retain the above copyright notice, this
	// list of conditions and the following disclaimer.
	//
	// 2. Redistributions in binary form must reproduce the above copyright notice,
	// this list of conditions and the following disclaimer in the documentation
	// and/or other materials provided with the distribution.
	//
	// 3. Neither the name of the copyright holder nor the names of its
	// contributors may be used to endorse or promote products derived from
	// this software without specific prior written permission.
	//
	// THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
	// AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
	// IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
	// DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS BE LIABLE
	// FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
	// DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR
	// SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER
	// CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY,
	// OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
	// OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.

	#include "src/tint/lang/wgsl/resolver/resolver.h"

	#include "gmock/gmock.h"
	#include "src/tint/lang/wgsl/resolver/resolver_helper_test.h"
	#include "src/tint/utils/containers/slice.h"

	namespace tint::resolver {
	namespace {

	using namespace tint::core::fluent_types; // NOLINT
	using namespace tint::core::number_suffixes; // NOLINT

	using ResolverEvaluationStageTest = ResolverTest;

	TEST_F(ResolverEvaluationStageTest, Literal_i32) {
	auto* expr = Expr(123_i);
	WrapInFunction(expr);

	ASSERT_TRUE(r()->Resolve()) << r()->error();
	EXPECT_EQ(Sem().Get(expr)->Stage(), core::EvaluationStage::kConstant);
	}

	TEST_F(ResolverEvaluationStageTest, Literal_f32) {
	auto* expr = Expr(123_f);
	WrapInFunction(expr);

	ASSERT_TRUE(r()->Resolve()) << r()->error();
	EXPECT_EQ(Sem().Get(expr)->Stage(), core::EvaluationStage::kConstant);
	}

	TEST_F(ResolverEvaluationStageTest, Vector_Init) {
	auto* expr = Call<vec3<f32>>();
	WrapInFunction(expr);

	ASSERT_TRUE(r()->Resolve()) << r()->error();
	EXPECT_EQ(Sem().Get(expr)->Stage(), core::EvaluationStage::kConstant);
	}

	TEST_F(ResolverEvaluationStageTest, Vector_Init_Const_Const) {
	// const f = 1.f;
	// vec2<f32>(f, f);
	auto* f = Const("f", Expr(1_f));
	auto* expr = Call<vec2<f32>>(f, f);
	WrapInFunction(f, expr);

	ASSERT_TRUE(r()->Resolve()) << r()->error();
	EXPECT_EQ(Sem().Get(f)->Stage(), core::EvaluationStage::kConstant);
	EXPECT_EQ(Sem().Get(expr)->Stage(), core::EvaluationStage::kConstant);
	}

	TEST_F(ResolverEvaluationStageTest, Vector_Init_Runtime_Runtime) {
	// var f = 1.f;
	// vec2<f32>(f, f);
	auto* f = Var("f", Expr(1_f));
	auto* expr = Call<vec2<f32>>(f, f);
	WrapInFunction(f, expr);

	ASSERT_TRUE(r()->Resolve()) << r()->error();
	EXPECT_EQ(Sem().Get(f)->Stage(), core::EvaluationStage::kRuntime);
	EXPECT_EQ(Sem().Get(expr)->Stage(), core::EvaluationStage::kRuntime);
	}

	TEST_F(ResolverEvaluationStageTest, Vector_Conv_Const) {
	// const f = 1.f;
	// vec2<u32>(vec2<f32>(f));
	auto* f = Const("f", Expr(1_f));
	auto* expr = Call<vec2<u32>>(Call<vec2<f32>>(f));
	WrapInFunction(f, expr);

	ASSERT_TRUE(r()->Resolve()) << r()->error();
	EXPECT_EQ(Sem().Get(f)->Stage(), core::EvaluationStage::kConstant);
	EXPECT_EQ(Sem().Get(expr)->Stage(), core::EvaluationStage::kConstant);
	}

	TEST_F(ResolverEvaluationStageTest, Vector_Conv_Runtime) {
	// var f = 1.f;
	// vec2<u32>(vec2<f32>(f));
	auto* f = Var("f", Expr(1_f));
	auto* expr = Call<vec2<u32>>(Call<vec2<f32>>(f));
	WrapInFunction(f, expr);

	ASSERT_TRUE(r()->Resolve()) << r()->error();
	EXPECT_EQ(Sem().Get(f)->Stage(), core::EvaluationStage::kRuntime);
	EXPECT_EQ(Sem().Get(expr)->Stage(), core::EvaluationStage::kRuntime);
	}

	TEST_F(ResolverEvaluationStageTest, Matrix_Init) {
	auto* expr = Call<mat2x2<f32>>();
	WrapInFunction(expr);

	ASSERT_TRUE(r()->Resolve()) << r()->error();
	EXPECT_EQ(Sem().Get(expr)->Stage(), core::EvaluationStage::kConstant);
	}

	TEST_F(ResolverEvaluationStageTest, Array_Init) {
	auto* expr = Call<array<f32, 3>>();
	WrapInFunction(expr);

	ASSERT_TRUE(r()->Resolve()) << r()->error();
	EXPECT_EQ(Sem().Get(expr)->Stage(), core::EvaluationStage::kConstant);
	}

	TEST_F(ResolverEvaluationStageTest, Array_Init_Const_Const) {
	// const f = 1.f;
	// array<f32, 2>(f, f);
	auto* f = Const("f", Expr(1_f));
	auto* expr = Call<array<f32, 2>>(f, f);
	WrapInFunction(f, expr);

	ASSERT_TRUE(r()->Resolve()) << r()->error();
	EXPECT_EQ(Sem().Get(f)->Stage(), core::EvaluationStage::kConstant);
	EXPECT_EQ(Sem().Get(expr)->Stage(), core::EvaluationStage::kConstant);
	}

	TEST_F(ResolverEvaluationStageTest, Array_Init_Const_Override) {
	// const f1 = 1.f;
	// override f2 = 2.f;
	// array<f32, 2>(f1, f2);
	auto* f1 = Const("f1", Expr(1_f));
	auto* f2 = Override("f2", Expr(2_f));
	auto* expr = Call<array<f32, 2>>(f1, f2);
	WrapInFunction(f1, expr);

	ASSERT_TRUE(r()->Resolve()) << r()->error();
	EXPECT_EQ(Sem().Get(f1)->Stage(), core::EvaluationStage::kConstant);
	EXPECT_EQ(Sem().Get(f2)->Stage(), core::EvaluationStage::kOverride);
	EXPECT_EQ(Sem().Get(expr)->Stage(), core::EvaluationStage::kOverride);
	}

	TEST_F(ResolverEvaluationStageTest, Array_Init_Override_Runtime) {
	// override f1 = 1.f;
	// var f2 = 2.f;
	// array<f32, 2>(f1, f2);
	auto* f1 = Override("f1", Expr(1_f));
	auto* f2 = Var("f2", Expr(2_f));
	auto* expr = Call<array<f32, 2>>(f1, f2);
	WrapInFunction(f2, expr);

	ASSERT_TRUE(r()->Resolve()) << r()->error();
	EXPECT_EQ(Sem().Get(f1)->Stage(), core::EvaluationStage::kOverride);
	EXPECT_EQ(Sem().Get(f2)->Stage(), core::EvaluationStage::kRuntime);
	EXPECT_EQ(Sem().Get(expr)->Stage(), core::EvaluationStage::kRuntime);
	}

	TEST_F(ResolverEvaluationStageTest, Array_Init_Const_Runtime) {
	// const f1 = 1.f;
	// var f2 = 2.f;
	// array<f32, 2>(f1, f2);
	auto* f1 = Const("f1", Expr(1_f));
	auto* f2 = Var("f2", Expr(2_f));
	auto* expr = Call<array<f32, 2>>(f1, f2);
	WrapInFunction(f1, f2, expr);

	ASSERT_TRUE(r()->Resolve()) << r()->error();
	EXPECT_EQ(Sem().Get(f1)->Stage(), core::EvaluationStage::kConstant);
	EXPECT_EQ(Sem().Get(f2)->Stage(), core::EvaluationStage::kRuntime);
	EXPECT_EQ(Sem().Get(expr)->Stage(), core::EvaluationStage::kRuntime);
	}

	TEST_F(ResolverEvaluationStageTest, Array_Init_Runtime_Runtime) {
	// var f = 1.f;
	// array<f32, 2>(f, f);
	auto* f = Var("f", Expr(1_f));
	auto* expr = Call<array<f32, 2>>(f, f);
	WrapInFunction(f, expr);

	ASSERT_TRUE(r()->Resolve()) << r()->error();
	EXPECT_EQ(Sem().Get(f)->Stage(), core::EvaluationStage::kRuntime);
	EXPECT_EQ(Sem().Get(expr)->Stage(), core::EvaluationStage::kRuntime);
	}

	TEST_F(ResolverEvaluationStageTest, IndexAccessor_Const_Const) {
	// const vec = vec4<f32>();
	// const idx = 1_i;
	// vec[idx]
	auto* vec = Const("vec", Call<vec4<f32>>());
	auto* idx = Const("idx", Expr(1_i));
	auto* expr = IndexAccessor(vec, idx);
	WrapInFunction(vec, idx, expr);

	ASSERT_TRUE(r()->Resolve()) << r()->error();
	EXPECT_EQ(Sem().Get(vec)->Stage(), core::EvaluationStage::kConstant);
	EXPECT_EQ(Sem().Get(idx)->Stage(), core::EvaluationStage::kConstant);
	EXPECT_EQ(Sem().Get(expr)->Stage(), core::EvaluationStage::kConstant);
	}

	TEST_F(ResolverEvaluationStageTest, IndexAccessor_Runtime_Const) {
	// var vec = vec4<f32>();
	// const idx = 1_i;
	// vec[idx]
	auto* vec = Var("vec", Call<vec4<f32>>());
	auto* idx = Const("idx", Expr(1_i));
	auto* expr = IndexAccessor(vec, idx);
	WrapInFunction(vec, idx, expr);

	ASSERT_TRUE(r()->Resolve()) << r()->error();
	EXPECT_EQ(Sem().Get(vec)->Stage(), core::EvaluationStage::kRuntime);
	EXPECT_EQ(Sem().Get(idx)->Stage(), core::EvaluationStage::kConstant);
	EXPECT_EQ(Sem().Get(expr)->Stage(), core::EvaluationStage::kRuntime);
	}

	TEST_F(ResolverEvaluationStageTest, IndexAccessor_Const_Override) {
	// const vec = vec4<f32>();
	// override idx = 1_i;
	// vec[idx]
	auto* vec = Const("vec", Call<vec4<f32>>());
	auto* idx = Override("idx", Expr(1_i));
	auto* expr = IndexAccessor(vec, idx);
	WrapInFunction(vec, expr);

	ASSERT_TRUE(r()->Resolve()) << r()->error();
	EXPECT_EQ(Sem().Get(vec)->Stage(), core::EvaluationStage::kConstant);
	EXPECT_EQ(Sem().Get(idx)->Stage(), core::EvaluationStage::kOverride);
	EXPECT_EQ(Sem().Get(expr)->Stage(), core::EvaluationStage::kOverride);
	}

	TEST_F(ResolverEvaluationStageTest, IndexAccessor_Const_Runtime) {
	// const vec = vec4<f32>();
	// let idx = 1_i;
	// vec[idx]
	auto* vec = Const("vec", Call<vec4<f32>>());
	auto* idx = Let("idx", Expr(1_i));
	auto* expr = IndexAccessor(vec, idx);
	WrapInFunction(vec, idx, expr);

	ASSERT_TRUE(r()->Resolve()) << r()->error();
	EXPECT_EQ(Sem().Get(vec)->Stage(), core::EvaluationStage::kConstant);
	EXPECT_EQ(Sem().Get(idx)->Stage(), core::EvaluationStage::kRuntime);
	EXPECT_EQ(Sem().Get(expr)->Stage(), core::EvaluationStage::kRuntime);
	}

	TEST_F(ResolverEvaluationStageTest, Swizzle_Const) {
	// const vec = S();
	// vec.m
	auto* vec = Const("vec", Call<vec4<f32>>());
	auto* expr = MemberAccessor(vec, "xz");
	WrapInFunction(vec, expr);

	ASSERT_TRUE(r()->Resolve()) << r()->error();
	EXPECT_EQ(Sem().Get(vec)->Stage(), core::EvaluationStage::kConstant);
	EXPECT_EQ(Sem().Get(expr)->Stage(), core::EvaluationStage::kConstant);
	}

	TEST_F(ResolverEvaluationStageTest, Swizzle_Runtime) {
	// var vec = S();
	// vec.m
	auto* vec = Var("vec", Call<vec4<f32>>());
	auto* expr = MemberAccessor(vec, "rg");
	WrapInFunction(vec, expr);

	ASSERT_TRUE(r()->Resolve()) << r()->error();
	EXPECT_EQ(Sem().Get(vec)->Stage(), core::EvaluationStage::kRuntime);
	EXPECT_EQ(Sem().Get(expr)->Stage(), core::EvaluationStage::kRuntime);
	}

	TEST_F(ResolverEvaluationStageTest, MemberAccessor_Const) {
	// struct S { m : i32 };
	// const str = S();
	// str.m
	Structure("S", Vector{Member("m", ty.i32())});
	auto* str = Const("str", Call("S"));
	auto* expr = MemberAccessor(str, "m");
	WrapInFunction(str, expr);

	ASSERT_TRUE(r()->Resolve()) << r()->error();
	EXPECT_EQ(Sem().Get(str)->Stage(), core::EvaluationStage::kConstant);
	EXPECT_EQ(Sem().Get(expr)->Stage(), core::EvaluationStage::kConstant);
	}

	TEST_F(ResolverEvaluationStageTest, MemberAccessor_Runtime) {
	// struct S { m : i32 };
	// var str = S();
	// str.m
	Structure("S", Vector{Member("m", ty.i32())});
	auto* str = Var("str", Call("S"));
	auto* expr = MemberAccessor(str, "m");
	WrapInFunction(str, expr);

	ASSERT_TRUE(r()->Resolve()) << r()->error();
	EXPECT_EQ(Sem().Get(str)->Stage(), core::EvaluationStage::kRuntime);
	EXPECT_EQ(Sem().Get(expr)->Stage(), core::EvaluationStage::kRuntime);
	}

	TEST_F(ResolverEvaluationStageTest, Binary_Runtime) {
	// let one = 1;
	// let result = (one == 1) && (one == 1);
	auto* one = Let("one", Expr(1_a));
	auto* lhs = Equal("one", 1_a);
	auto* rhs = Equal("one", 1_a);
	auto* binary = LogicalAnd(lhs, rhs);
	auto* result = Let("result", binary);
	WrapInFunction(one, result);

	ASSERT_TRUE(r()->Resolve()) << r()->error();
	EXPECT_EQ(Sem().Get(lhs)->Stage(), core::EvaluationStage::kRuntime);
	EXPECT_EQ(Sem().Get(rhs)->Stage(), core::EvaluationStage::kRuntime);
	EXPECT_EQ(Sem().Get(binary)->Stage(), core::EvaluationStage::kRuntime);
	}

	TEST_F(ResolverEvaluationStageTest, Binary_Const) {
	// const one = 1;
	// const result = (one == 1) && (one == 1);
	auto* one = Const("one", Expr(1_a));
	auto* lhs = Equal("one", 1_a);
	auto* rhs = Equal("one", 1_a);
	auto* binary = LogicalAnd(lhs, rhs);
	auto* result = Const("result", binary);
	WrapInFunction(one, result);

	ASSERT_TRUE(r()->Resolve()) << r()->error();
	EXPECT_EQ(Sem().Get(lhs)->Stage(), core::EvaluationStage::kConstant);
	EXPECT_EQ(Sem().Get(rhs)->Stage(), core::EvaluationStage::kConstant);
	EXPECT_EQ(Sem().Get(binary)->Stage(), core::EvaluationStage::kConstant);
	}

	TEST_F(ResolverEvaluationStageTest, Binary_NotEvaluated) {
	// const one = 1;
	// const result = (one == 0) && (one == 1);
	auto* one = Const("one", Expr(1_a));
	auto* lhs = Equal("one", 0_a);
	auto* rhs = Equal("one", 1_a);
	auto* binary = LogicalAnd(lhs, rhs);
	auto* result = Const("result", binary);
	WrapInFunction(one, result);

	ASSERT_TRUE(r()->Resolve()) << r()->error();
	EXPECT_EQ(Sem().Get(lhs)->Stage(), core::EvaluationStage::kConstant);
	EXPECT_EQ(Sem().Get(rhs)->Stage(), core::EvaluationStage::kNotEvaluated);
	EXPECT_EQ(Sem().Get(binary)->Stage(), core::EvaluationStage::kConstant);
	}

	TEST_F(ResolverEvaluationStageTest, FnCall_Runtime) {
	// fn f() -> bool { return true; }
	// let l = false
	// let result = l && f();
	Func("f", Empty, ty.bool_(), Vector{Return(true)});
	auto* let = Let("l", Expr(false));
	auto* lhs = Expr(let);
	auto* rhs = Call("f");
	auto* binary = LogicalAnd(lhs, rhs);
	auto* result = Let("result", binary);
	WrapInFunction(let, result);

	ASSERT_TRUE(r()->Resolve()) << r()->error();
	EXPECT_EQ(Sem().Get(rhs)->Stage(), core::EvaluationStage::kRuntime);
	EXPECT_EQ(Sem().Get(binary)->Stage(), core::EvaluationStage::kRuntime);
	}

	TEST_F(ResolverEvaluationStageTest, FnCall_NotEvaluated) {
	// fn f() -> bool { return true; }
	// let result = false && f();
	Func("f", Empty, ty.bool_(), Vector{Return(true)});
	auto* rhs = Call("f");
	auto* lhs = Expr(false);
	auto* binary = LogicalAnd(lhs, rhs);
	auto* result = Let("result", binary);
	WrapInFunction(result);

	ASSERT_TRUE(r()->Resolve()) << r()->error();
	EXPECT_EQ(Sem().Get(rhs)->Stage(), core::EvaluationStage::kNotEvaluated);
	EXPECT_EQ(Sem().Get(binary)->Stage(), core::EvaluationStage::kConstant);
	}

	TEST_F(ResolverEvaluationStageTest, NestedFnCall_NotEvaluated) {
	// fn f(b : bool) -> bool { return b; }
	// let result = false && f(f(f(1 == 0)));
	Func("f", Vector{Param("b", ty.bool_())}, ty.bool_(), Vector{Return("b")});
	auto* cmp = Equal(0_i, 1_i);
	auto* rhs_0 = Call("f", cmp);
	auto* rhs_1 = Call("f", rhs_0);
	auto* rhs_2 = Call("f", rhs_1);
	auto* lhs = Expr(false);
	auto* binary = LogicalAnd(lhs, rhs_2);
	auto* result = Let("result", binary);
	WrapInFunction(result);

	ASSERT_TRUE(r()->Resolve()) << r()->error();
	EXPECT_EQ(Sem().Get(cmp)->Stage(), core::EvaluationStage::kNotEvaluated);
	EXPECT_EQ(Sem().Get(rhs_0)->Stage(), core::EvaluationStage::kNotEvaluated);
	EXPECT_EQ(Sem().Get(rhs_1)->Stage(), core::EvaluationStage::kNotEvaluated);
	EXPECT_EQ(Sem().Get(rhs_2)->Stage(), core::EvaluationStage::kNotEvaluated);
	EXPECT_EQ(Sem().Get(binary)->Stage(), core::EvaluationStage::kConstant);
	}

	} // namespace
	} // namespace tint::resolver