Skip to content

Commit

Permalink
Remove unsupported ARCH::WORMHOLE
Browse files Browse the repository at this point in the history
  • Loading branch information
broskoTT committed Oct 31, 2024
1 parent f879f34 commit 610750e
Show file tree
Hide file tree
Showing 4 changed files with 6 additions and 6 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -686,7 +686,7 @@ uint32_t get_l1_size(tt::ARCH arch) {
constexpr uint32_t WH_L1_SIZE = 1499136;

uint32_t l1_size = 0;
if (arch == tt::ARCH::WORMHOLE || arch == tt::ARCH::WORMHOLE_B0) {
if (arch == tt::ARCH::WORMHOLE_B0) {
l1_size = WH_L1_SIZE;
} else if (arch == tt::ARCH::GRAYSKULL) {
l1_size = GS_L1_SIZE;
Expand All @@ -701,7 +701,7 @@ double get_tt_npu_rpeak_tflops(tt::ARCH arch, CoreCoord grid_size, int tt_npu_cl
double rpeak_tflops = 0.0f;
double clock = static_cast<double>(tt_npu_clock) / 1000;
uint32_t num_compute_core = grid_size.x * grid_size.y;
if (arch == tt::ARCH::WORMHOLE || arch == tt::ARCH::WORMHOLE_B0) {
if (arch == tt::ARCH::WORMHOLE_B0) {
rpeak_tflops =
WH_FPU_BFP8_TFLOPS_PER_TENSIX * static_cast<double>(num_compute_core) * static_cast<double>(clock);
} else if (arch == tt::ARCH::GRAYSKULL) {
Expand Down Expand Up @@ -777,7 +777,7 @@ CoreCoord get_core_range(
std::tuple<MathFidelity, bool> get_compute_params(tt::ARCH arch) {
MathFidelity math_fidelity = MathFidelity::HiFi4;
bool fp32_dest_acc_en = false;
if (arch == tt::ARCH::WORMHOLE || arch == tt::ARCH::WORMHOLE_B0) {
if (arch == tt::ARCH::WORMHOLE_B0) {
math_fidelity = MathFidelity::HiFi2;
// TODO: apply packer_l1_acc
// TODO: need to consider whether to set these variablias as arguments
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -514,7 +514,7 @@ uint32_t get_dram_bandwidth(tt::ARCH arch) {
constexpr uint32_t WH_DRAM_BANDWIDTH_GB_PER_SEC = 384;

uint32_t dram_bandwidth_gb_per_sec = 0;
if (arch == tt::ARCH::WORMHOLE || arch == tt::ARCH::WORMHOLE_B0) {
if (arch == tt::ARCH::WORMHOLE_B0) {
dram_bandwidth_gb_per_sec = WH_DRAM_BANDWIDTH_GB_PER_SEC;
} else if (arch == tt::ARCH::GRAYSKULL) {
dram_bandwidth_gb_per_sec = GS_DRAM_BANDWIDTH_GB_PER_SEC;
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -224,7 +224,7 @@ uint32_t get_dram_bandwidth(tt::ARCH arch) {
constexpr uint32_t WH_DRAM_BANDWIDTH_GB_PER_SEC = 384;

uint32_t dram_bandwidth_gb_per_sec = 0;
if (arch == tt::ARCH::WORMHOLE || arch == tt::ARCH::WORMHOLE_B0) {
if (arch == tt::ARCH::WORMHOLE_B0) {
dram_bandwidth_gb_per_sec = WH_DRAM_BANDWIDTH_GB_PER_SEC;
} else if (arch == tt::ARCH::GRAYSKULL) {
dram_bandwidth_gb_per_sec = GS_DRAM_BANDWIDTH_GB_PER_SEC;
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -258,7 +258,7 @@ uint32_t get_dram_bandwidth(tt::ARCH arch) {
constexpr uint32_t WH_DRAM_BANDWIDTH_GB_PER_SEC = 384;

uint32_t dram_bandwidth_gb_per_sec = 0;
if (arch == tt::ARCH::WORMHOLE || arch == tt::ARCH::WORMHOLE_B0) {
if (arch == tt::ARCH::WORMHOLE_B0) {
dram_bandwidth_gb_per_sec = WH_DRAM_BANDWIDTH_GB_PER_SEC;
} else if (arch == tt::ARCH::GRAYSKULL) {
dram_bandwidth_gb_per_sec = GS_DRAM_BANDWIDTH_GB_PER_SEC;
Expand Down

0 comments on commit 610750e

Please sign in to comment.