Skip to content

Commit

Permalink
#3561: Modify recip unit test for wormhole
Browse files Browse the repository at this point in the history
  • Loading branch information
nemanjagrujic committed Nov 2, 2023
1 parent 4c40e06 commit 1a34a01
Show file tree
Hide file tree
Showing 2 changed files with 52 additions and 3 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -5,12 +5,10 @@
from loguru import logger
import pytest
import torch

import tt_lib as ttl

from tests.tt_eager.python_api_testing.sweep_tests import pytorch_ops
from tests.tt_eager.python_api_testing.sweep_tests.comparison_funcs import comp_pcc
from tests.tt_eager.python_api_testing.sweep_tests.common import is_wormhole_b0, skip_for_wormhole_b0
from tests.tt_eager.python_api_testing.sweep_tests.tt_lib_ops import eltwise_recip as tt_eltwise_recip


Expand Down Expand Up @@ -78,7 +76,6 @@ def run_eltwise_recip_tests(input_shape, dtype, dlayout, in_mem_config, out_mem_
]


@skip_for_wormhole_b0
@pytest.mark.parametrize(
"input_shape, dtype, dlayout, in_mem_config, out_mem_config, data_seed",
(test_sweep_args),
Expand Down
Original file line number Diff line number Diff line change
@@ -0,0 +1,52 @@
---
test-list:
- eltwise-recip:
shape:
start-shape: [1, 1, 32, 32]
end-shape: [6, 12, 256, 256]
interval: [1, 1, 32, 32]
num-shapes: 1
num-samples: 64
args-sampling-strategy: "all"
env:
TT_METAL_SLOW_DISPATCH_MODE: ["", "1"]
# TT_PCI_DMA_BUF_SIZE: "1048576"
datagen:
function: gen_rand
args:
low: -100
high: 100
comparison:
function: comp_pcc
args-gen: gen_dtype_layout_device
args:
data-layout: ["TILE"]
data-type: ["BFLOAT16"]
buffer-type: ["DRAM", "L1", "SYSTEM_MEMORY"]
out-buffer-type: ["DRAM"]
output-file: eltwise_recip_sweep.csv
- eltwise-recip:
shape:
start-shape: [1, 1, 2, 2]
end-shape: [6, 12, 256, 256]
interval: [1, 1, 1, 2]
num-shapes: 1
num-samples: 64
args-sampling-strategy: "all"
env:
TT_METAL_SLOW_DISPATCH_MODE: ["", "1"]
# TT_PCI_DMA_BUF_SIZE: "1048576"
datagen:
function: gen_rand
args:
low: -100
high: 100
comparison:
function: comp_pcc
args-gen: gen_dtype_layout_device
args:
data-layout: ["ROW_MAJOR"]
data-type: ["BFLOAT16"]
buffer-type: ["DRAM", "L1", "SYSTEM_MEMORY"]
out-buffer-type: ["DRAM"]
output-file: eltwise_recip_sweep.csv

0 comments on commit 1a34a01

Please sign in to comment.