diff --git a/ttnn/cpp/ttnn/tensor/enum_types.hpp b/ttnn/cpp/ttnn/tensor/enum_types.hpp new file mode 100644 index 00000000000..9f9b0738e4c --- /dev/null +++ b/ttnn/cpp/ttnn/tensor/enum_types.hpp @@ -0,0 +1,11 @@ +// SPDX-FileCopyrightText: © 2024 Tenstorrent AI ULC +// +// SPDX-License-Identifier: Apache-2.0 + +#pragma once + +namespace tt::tt_metal { + +enum class Layout { ROW_MAJOR = 0, TILE = 1, INVALID = 2 }; + +} // namespace tt::tt_metal diff --git a/ttnn/cpp/ttnn/tensor/types.hpp b/ttnn/cpp/ttnn/tensor/types.hpp index 45544510e5a..84e9a97b291 100644 --- a/ttnn/cpp/ttnn/tensor/types.hpp +++ b/ttnn/cpp/ttnn/tensor/types.hpp @@ -17,6 +17,7 @@ #include "tt_metal/tt_stl/concepts.hpp" #include "tt_metal/tt_stl/reflection.hpp" #include "ttnn/tensor/host_buffer/types.hpp" +#include "ttnn/cpp/ttnn/tensor/enum_types.hpp" namespace tt { @@ -24,8 +25,6 @@ namespace tt_metal { static constexpr std::uint8_t VERSION_ID = 3; -enum class Layout { ROW_MAJOR = 0, TILE = 1, INVALID = 2 }; - enum class DataType { BFLOAT16 = 0, FLOAT32 = 1,