Skip to content

Commit

Permalink
#5389: removed early return from validate when enable_fast_runtime_mo…
Browse files Browse the repository at this point in the history
…de was set to true
  • Loading branch information
arakhmati committed Jun 3, 2024
1 parent 8e3cd8f commit fdabcb2
Show file tree
Hide file tree
Showing 3 changed files with 4 additions and 4 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -2,6 +2,8 @@

# SPDX-License-Identifier: Apache-2.0

import pytest

import torch
import ttnn
import tt_lib
Expand All @@ -19,6 +21,7 @@
from tests.ttnn.utils_for_testing import assert_with_pcc


@pytest.mark.skip(reason="https://github.com/tenstorrent/tt-metal/issues/9076")
@skip_for_wormhole_b0()
def test_mistral_attention_inference(model_location_generator, device, reset_seeds):
model_path = model_location_generator("mistral-7B-v0.1", model_subdir="Mistral")
Expand Down
3 changes: 0 additions & 3 deletions tt_eager/tt_dnn/op_library/operation.hpp
Original file line number Diff line number Diff line change
Expand Up @@ -528,9 +528,6 @@ struct DeviceOperation final {
const Tensors& input_tensors,
const OptionalConstTensors& optional_input_tensors,
const OptionalTensors& optional_output_tensors) -> void {
if (ttnn::CONFIG.enable_fast_runtime_mode) {
return;
}
const auto& operation = *reinterpret_cast<const std::decay_t<T>*>(&storage);
if constexpr (
(detail::implements_validate<T>() or
Expand Down
2 changes: 1 addition & 1 deletion ttnn/ttnn/__init__.py
Original file line number Diff line number Diff line change
Expand Up @@ -57,7 +57,7 @@ def validate(self, name):
if self.enable_fast_runtime_mode:
if self.enable_logging:
logger.warning(
"Running in fast runtime mode without logging. Please disable fast runtime mode if you want to enable logging."
"Logging cannot be enabled in fast runtime mode. Please disable fast runtime mode if you want to enable logging."
)

if name in {
Expand Down

0 comments on commit fdabcb2

Please sign in to comment.