Skip to content

Commit

Permalink
#8598: sinh fix (#12056)
Browse files Browse the repository at this point in the history
  • Loading branch information
ruthreshx authored Aug 30, 2024
1 parent 62d92cf commit 5235c85
Show file tree
Hide file tree
Showing 2 changed files with 3 additions and 3 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -22,7 +22,7 @@ def run_eltwise_sinh_tests(
device,
):
torch.manual_seed(data_seed)
x = torch.Tensor(size=input_shape[0]).uniform_(-100, 100).to(torch.bfloat16)
x = torch.Tensor(size=input_shape[0]).uniform_(-88, 88).to(torch.bfloat16)

try:
# get ref result
Expand Down Expand Up @@ -53,7 +53,7 @@ def run_eltwise_sinh_tests(
for dtype in [ttnn.bfloat16, ttnn.bfloat8_b]:
for layout in [ttnn.TILE_LAYOUT, ttnn.ROW_MAJOR_LAYOUT]:
for mem_cfg in [ttnn.DRAM_MEMORY_CONFIG, ttnn.L1_MEMORY_CONFIG]:
if layout == ttnn.ROW_MAJOR_LAYOUT and dtype == ttnn.bfloat8_b:
if layout == ttnn.ROW_MAJOR_LAYOUT or dtype == ttnn.bfloat8_b:
continue

test_sweep_args.append(
Expand Down
2 changes: 1 addition & 1 deletion ttnn/cpp/ttnn/operations/eltwise/unary/unary_pybind.hpp
Original file line number Diff line number Diff line change
Expand Up @@ -1351,7 +1351,7 @@ void py_module(py::module& module) {
detail::bind_unary_composite(module, ttnn::log1p, R"doc(Performs log1p function on :attr:`input_tensor`.)doc", "[supported range -1 to 1]");
detail::bind_unary_composite(module, ttnn::mish, R"doc(Performs mish function on :attr:`input_tensor`, not supported for grayskull.)doc");
detail::bind_unary_composite(module, ttnn::multigammaln, R"doc(Performs multigammaln function on :attr:`input_tensor`.)doc", "[supported range 1.6 to inf]");
detail::bind_unary_composite(module, ttnn::sinh, R"doc(Performs sinh function on :attr:`input_tensor`.)doc", "[supported range -9 to 9]");
detail::bind_unary_composite(module, ttnn::sinh, R"doc(Performs sinh function on :attr:`input_tensor`.)doc", "[supported range -88 to 88]");
detail::bind_unary_composite(module, ttnn::softsign, R"doc(Performs softsign function on :attr:`input_tensor`.)doc");
detail::bind_unary_composite(module, ttnn::swish, R"doc(Performs swish function on :attr:`input_tensor`.)doc");
detail::bind_unary_composite(module, ttnn::trunc, R"doc(Performs trunc function on :attr:`input_tensor`, not supported for grayskull.)doc");
Expand Down

0 comments on commit 5235c85

Please sign in to comment.