Skip to content

Commit

Permalink
#4702: Update layout for gelu test
Browse files Browse the repository at this point in the history
  • Loading branch information
umadevimcw committed Nov 12, 2024
1 parent 49ba42d commit 4cc52b4
Showing 1 changed file with 3 additions and 3 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -40,23 +40,23 @@ def run_eltwise_gelu_tests(input_shape, dtype, dlayout, in_mem_config, output_me
(
[(150, 72)],
[ttnn.bfloat16],
[ttnn.ROW_MAJOR_LAYOUT],
[ttnn.TILE_LAYOUT],
(ttnn.DRAM_MEMORY_CONFIG),
(ttnn.DRAM_MEMORY_CONFIG),
17799073,
),
(
[(3, 201, 228)],
[ttnn.bfloat16],
[ttnn.ROW_MAJOR_LAYOUT],
[ttnn.TILE_LAYOUT],
(ttnn.DRAM_MEMORY_CONFIG),
(ttnn.DRAM_MEMORY_CONFIG),
3121221,
),
(
[(6, 6, 230, 138)],
[ttnn.bfloat16],
[ttnn.ROW_MAJOR_LAYOUT],
[ttnn.TILE_LAYOUT],
(ttnn.DRAM_MEMORY_CONFIG),
(ttnn.DRAM_MEMORY_CONFIG),
10286194,
Expand Down

0 comments on commit 4cc52b4

Please sign in to comment.