| // Copyright 2023 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/constant/manager.h" |
| |
| #include "src/tint/constant/composite.h" |
| #include "src/tint/constant/scalar.h" |
| #include "src/tint/constant/splat.h" |
| #include "src/tint/type/abstract_float.h" |
| #include "src/tint/type/abstract_int.h" |
| #include "src/tint/type/bool.h" |
| #include "src/tint/type/f16.h" |
| #include "src/tint/type/f32.h" |
| #include "src/tint/type/i32.h" |
| #include "src/tint/type/manager.h" |
| #include "src/tint/type/u32.h" |
| #include "src/tint/utils/predicates.h" |
| |
| namespace tint::constant { |
| |
| Manager::Manager() = default; |
| |
| Manager::Manager(Manager&&) = default; |
| |
| Manager& Manager::operator=(Manager&& rhs) = default; |
| |
| Manager::~Manager() = default; |
| |
| const constant::Value* Manager::Composite(const type::Type* type, |
| utils::VectorRef<const constant::Value*> elements) { |
| if (elements.IsEmpty()) { |
| return nullptr; |
| } |
| |
| bool any_zero = false; |
| bool all_zero = true; |
| bool all_equal = true; |
| auto* first = elements.Front(); |
| for (auto* el : elements) { |
| if (!el) { |
| return nullptr; |
| } |
| if (!any_zero && el->AnyZero()) { |
| any_zero = true; |
| } |
| if (all_zero && !el->AllZero()) { |
| all_zero = false; |
| } |
| if (all_equal && el != first) { |
| all_equal = false; |
| } |
| } |
| if (all_equal) { |
| return Splat(type, elements.Front(), elements.Length()); |
| } |
| |
| return Get<constant::Composite>(type, std::move(elements), all_zero, any_zero); |
| } |
| |
| const constant::Splat* Manager::Splat(const type::Type* type, |
| const constant::Value* element, |
| size_t n) { |
| return Get<constant::Splat>(type, element, n); |
| } |
| |
| const Scalar<i32>* Manager::Get(i32 value) { |
| return Get<Scalar<i32>>(types.i32(), value); |
| } |
| |
| const Scalar<u32>* Manager::Get(u32 value) { |
| return Get<Scalar<u32>>(types.u32(), value); |
| } |
| |
| const Scalar<f32>* Manager::Get(f32 value) { |
| return Get<Scalar<f32>>(types.f32(), value); |
| } |
| |
| const Scalar<f16>* Manager::Get(f16 value) { |
| return Get<Scalar<f16>>(types.f16(), value); |
| } |
| |
| const Scalar<bool>* Manager::Get(bool value) { |
| return Get<Scalar<bool>>(types.bool_(), value); |
| } |
| |
| const Scalar<AFloat>* Manager::Get(AFloat value) { |
| return Get<Scalar<AFloat>>(types.AFloat(), value); |
| } |
| |
| const Scalar<AInt>* Manager::Get(AInt value) { |
| return Get<Scalar<AInt>>(types.AInt(), value); |
| } |
| |
| } // namespace tint::constant |