Skip to content

Commit

Permalink
#0: update model performance
Browse files Browse the repository at this point in the history
- update license years
  • Loading branch information
sjameelTT committed Nov 19, 2024
1 parent e279bf0 commit 91b1079
Show file tree
Hide file tree
Showing 6 changed files with 10 additions and 7 deletions.
2 changes: 1 addition & 1 deletion models/demos/distilbert/tests/test_perf_distilbert.py
Original file line number Diff line number Diff line change
Expand Up @@ -154,7 +154,7 @@ def test_distilbert_perf_device(batch_size, test, reset_seeds):
if is_grayskull():
expected_perf = 40.8772
elif is_wormhole_b0():
expected_perf = 90.2505
expected_perf = 103.884

command = f"pytest tests/ttnn/integration_tests/distilbert/test_ttnn_distilbert.py::test_distilbert_for_question_answering[sequence_size=768-batch_size=8-model_name=distilbert-base-uncased-distilled-squad]"
cols = ["DEVICE FW", "DEVICE KERNEL", "DEVICE BRISC KERNEL"]
Expand Down
4 changes: 2 additions & 2 deletions models/demos/vgg/tests/test_perf_vgg.py
Original file line number Diff line number Diff line change
Expand Up @@ -137,10 +137,10 @@ def test_perf_device_bare_metal_vgg(batch_size, model_name):
margin = 0.03

if model_name == "ttnn_vgg11":
expected_perf = 132.2436 if is_grayskull() else 272.8989
expected_perf = 132.2436 if is_grayskull() else 283.289
command = f"pytest tests/ttnn/integration_tests/vgg/test_ttnn_vgg11.py"
else:
expected_perf = 116.1459 if is_grayskull() else 194.4063
expected_perf = 116.1459 if is_grayskull() else 201.3867
command = f"pytest tests/ttnn/integration_tests/vgg/test_ttnn_vgg16.py"

cols = ["DEVICE FW", "DEVICE KERNEL", "DEVICE BRISC KERNEL"]
Expand Down
Original file line number Diff line number Diff line change
@@ -1,4 +1,4 @@
# SPDX-FileCopyrightText: © 2023 Tenstorrent Inc.
# SPDX-FileCopyrightText: © 2024 Tenstorrent Inc.

# SPDX-License-Identifier: Apache-2.0

Expand Down Expand Up @@ -911,6 +911,8 @@ def test_transpose_issue_11650_10350(shape, dims, layout, dtype, device):
def test_transpose_unpadded(shape, dims, layout, dtype, pad_value, device):
if pad_value is not None and is_blackhole():
pytest.skip("Blackhole reduce is needed for the full test to work")
elif dtype == ttnn.float32 and is_grayskull():
pytest.skip("Grayskull does not support float32")
torch_input = torch.randn(shape, dtype=torch.bfloat16)
torch_output = torch_input.transpose(dims[0], dims[1])

Expand Down
3 changes: 2 additions & 1 deletion tests/ttnn/unit_tests/operations/test_permute.py
Original file line number Diff line number Diff line change
Expand Up @@ -8,7 +8,8 @@

import ttnn

from tests.ttnn.utils_for_testing import assert_with_pcc, is_blackhole
from tests.ttnn.utils_for_testing import assert_with_pcc
from models.utility_functions import is_blackhole

torch.manual_seed(2005)

Expand Down
2 changes: 1 addition & 1 deletion ttnn/cpp/ttnn/operations/data_movement/concat/concat.cpp
Original file line number Diff line number Diff line change
Expand Up @@ -161,7 +161,7 @@ MassagedConcat build_prepost_transpose_concat(uint8_t queue_id, const MemoryConf
tensors.end(),
std::back_inserter(itensors),
[dim1, dim2](const ttnn::Tensor& input_tensor) -> ttnn::Tensor {
return ttnn::transpose(input_tensor, dim1, dim2, std::nullopt);
return ttnn::transpose(input_tensor, dim1, dim2);
}
);
auto norm_dim1 = tensors.front().get_shape().get_normalized_index(dim1);
Expand Down
Original file line number Diff line number Diff line change
@@ -1,4 +1,4 @@
// SPDX-FileCopyrightText: © 2023 Tenstorrent Inc.
// SPDX-FileCopyrightText: © 2024 Tenstorrent Inc.
//
// SPDX-License-Identifier: Apache-2.0

Expand Down

0 comments on commit 91b1079

Please sign in to comment.