((float *)mem_00)[dst_off_f(p, mb, oc)] = value;
});
- mem_dt.reorder(mem_00);
- mem_fp.reorder(mem_dt);
-
SAFE(mem_dt.reorder(mem_00), WARN);
SAFE(mem_fp.reorder(mem_dt), WARN);
if (bench_mode & CORR) {
compute_ref_fwd(p, src_fp, wei_fp, bia_fp, dst_fp);
dnn_mem_t dst(dst_dt, fp, mkldnn_nc);
- SAFE(dst.reorder(dst_dt), WARN);
SAFE(compare_dat(p, DST, dst, dst_fp, r), WARN);
}
} else if (p->dir == BWD_D) {
if (bench_mode & CORR) {
compute_ref_bwd_d(p, src_fp, wei_fp, dst_fp);
dnn_mem_t src(src_dt, fp, src_format);
- SAFE(src.reorder(src_dt), WARN);
SAFE(compare_dat(p, SRC, src, src_fp, r), WARN);
}
} else if (p->dir & FLAG_BWD && p->dir & FLAG_WEI) {
if (bench_mode & CORR) {
compute_ref_bwd_w(p, src_fp, wei_fp, bia_fp, dst_fp);
dnn_mem_t wei(wei_dt, fp, wei_format);
- SAFE(wei.reorder(wei_dt), WARN);
if (compare_dat(p, WEI, wei, wei_fp, r) != OK) return FAIL;
if (p->dir & FLAG_BIA) {
dnn_mem_t bia(bia_dt, fp, mkldnn_x);
- SAFE(bia.reorder(bia_dt), WARN);
SAFE(compare_dat(p, BIA, bia, bia_fp, r), WARN);
}
}