From 2d4fd76ca39092ca5dec23912d137695b34f1e98 Mon Sep 17 00:00:00 2001 From: pjanevski Date: Mon, 2 Dec 2024 11:14:55 +0000 Subject: [PATCH] Remove core type worker --- device/api/umd/device/tt_core_coordinates.h | 6 ++---- device/tt_soc_descriptor.cpp | 2 +- 2 files changed, 3 insertions(+), 5 deletions(-) diff --git a/device/api/umd/device/tt_core_coordinates.h b/device/api/umd/device/tt_core_coordinates.h index 8235db2f..01b722bd 100644 --- a/device/api/umd/device/tt_core_coordinates.h +++ b/device/api/umd/device/tt_core_coordinates.h @@ -16,8 +16,7 @@ * CoreType is an enum class that represents all types of cores * present on the Tenstorrent chip. */ -// TODO: change to uint8_t and uplift to tt-metal -enum class CoreType { +enum class CoreType : uint8_t { ARC, DRAM, ACTIVE_ETH, @@ -29,14 +28,13 @@ enum class CoreType { // but it won't be needed later on HARVESTED, ETH, - WORKER, }; /* * CoordSystem is an enum class that represents all types of coordinate * systems that can be used to represent a core's location. */ -enum class CoordSystem : std::uint8_t { +enum class CoordSystem : uint8_t { LOGICAL, PHYSICAL, VIRTUAL, diff --git a/device/tt_soc_descriptor.cpp b/device/tt_soc_descriptor.cpp index b8e61088..45d15972 100644 --- a/device/tt_soc_descriptor.cpp +++ b/device/tt_soc_descriptor.cpp @@ -131,7 +131,7 @@ void tt_SocDescriptor::load_core_descriptors_from_device_descriptor(YAML::Node & for (const auto &core_string : worker_cores) { CoreDescriptor core_descriptor; core_descriptor.coord = format_node(core_string); - core_descriptor.type = CoreType::WORKER; + core_descriptor.type = CoreType::TENSIX; core_descriptor.l1_size = worker_l1_size; cores.insert({core_descriptor.coord, core_descriptor}); workers.push_back(core_descriptor.coord);