Skip to content

Commit d1d653b

Browse files
author
Olivia W Hsu
committed
Fix medical imaging parameters for lanka
1 parent a60d288 commit d1d653b

File tree

3 files changed

+10
-8
lines changed

3 files changed

+10
-8
lines changed

numpy/image.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -116,7 +116,7 @@ def dense_bench():
116116
#TODO: Add in a benchmark that uses windowing for medical imaging as well.
117117
@pytest.mark.parametrize("num", list(range(1, 99)))
118118
@pytest.mark.parametrize("pt1", [0.75])
119-
@pytest.mark.parametrize("window_size", [0.25, 0.2, 0.15, 0.1])
119+
@pytest.mark.parametrize("window_size", [0.45, 0.4, 0.35, 0.3])
120120
def bench_edge_detection_window_pydata(tacoBench, num, pt1, window_size, plot):
121121
loader = ImagePydataSparseTensorLoader()
122122
sparse_bin_img1 = safeCastPydataTensorToInts(loader.sparse_image(num, pt1, 1))
@@ -169,7 +169,7 @@ def dense_bench():
169169

170170
@pytest.mark.parametrize("num", list(range(1, 99)))
171171
@pytest.mark.parametrize("pt1", [0.75])
172-
@pytest.mark.parametrize("window_size", [0.25, 0.2, 0.15, 0.1])
172+
@pytest.mark.parametrize("window_size", [0.45, 0.4, 0.35, 0.3])
173173
def bench_edge_detection_window_dense(tacoBench, num, pt1, window_size):
174174
loader = ImagePydataSparseTensorLoader()
175175
bin_img1 = loader.dense_image(num, pt1, 1)
@@ -264,7 +264,7 @@ def sparse_bench():
264264
@pytest.mark.skip(reasoun="For image generation only")
265265
@pytest.mark.parametrize("num", [42, 44, 50, 63, 92])
266266
@pytest.mark.parametrize("pt1", [0.75])
267-
def bench_edge_detection_fused_pydata(tacoBench, num, pt1, plot):
267+
def bench_edge_detection_plot(tacoBench, num, pt1, plot):
268268
loader = ImagePydataSparseTensorLoader()
269269
sparse_bin_img1 = safeCastPydataTensorToInts(loader.sparse_image(num, pt1, 1))
270270
sparse_bin_img2 = safeCastPydataTensorToInts(loader.sparse_image(num, pt1+0.05, 2))

scripts/bench_csv_aggregator.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -36,11 +36,13 @@ def aggregateTacoBenches(folder, outfile, taco=False, labelSet=None):
3636
# entries that have a skip marker in the label.
3737
# labelIdx = header.index("label", 0)
3838
if first:
39+
header.append("original_filename")
3940
writer.writerow(header)
4041
first = False
4142
for row in reader:
4243
# if "SKIPPED" not in row[labelIdx]:
4344
# validLabels.add(row[labelIdx])
45+
row.append(fname)
4446
writer.writerow(row)
4547
# Write out the set of valid labels.
4648
if labelSet is not None:

taco/image.cpp

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -114,7 +114,7 @@ TACO_BENCH_ARGS(bench_image_xor, csr, CSR);//->Apply(CustomArguments);
114114
static void bench_image_fused(benchmark::State& state, const Format& f) {
115115
// int num = state.range(0);
116116
auto t1 = 0.75;
117-
auto t2 = 0.55;
117+
auto t2 = 0.80;
118118

119119
auto num_str = getEnvVar("IMAGE_NUM");
120120
if (num_str == "") {
@@ -242,7 +242,7 @@ static void bench_image_window(benchmark::State& state, const Format& f, double
242242
// codegen->compile(compute, true);
243243
}
244244
}
245-
TACO_BENCH_ARGS(bench_image_window, csr/0.25, CSR, 0.25);
246-
TACO_BENCH_ARGS(bench_image_window, csr/0.2, CSR, 0.2);
247-
TACO_BENCH_ARGS(bench_image_window, csr/0.15, CSR, 0.15);
248-
TACO_BENCH_ARGS(bench_image_window, csr/0.1, CSR, 0.1);
245+
TACO_BENCH_ARGS(bench_image_window, csr/0.45, CSR, 0.45);
246+
TACO_BENCH_ARGS(bench_image_window, csr/0.4, CSR, 0.4);
247+
TACO_BENCH_ARGS(bench_image_window, csr/0.35, CSR, 0.35);
248+
TACO_BENCH_ARGS(bench_image_window, csr/0.3, CSR, 0.3);

0 commit comments

Comments
 (0)