Skip to content

Commit 609a583

Browse files
committed
throw exception on unbroadcastable input
1 parent 4e61135 commit 609a583

File tree

1 file changed

+6
-0
lines changed

1 file changed

+6
-0
lines changed

torch_ipex/csrc/cpu/DevOPs.cpp

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -471,6 +471,8 @@ at::Tensor& AtenIpexCPUDev::dil_baddbmm_out(
471471
if (!result.sizes().equals(inferred_size)) {
472472
result.resize_(inferred_size);
473473
}
474+
TORCH_CHECK(self.sizes().equals(inferred_size),
475+
"baddbmm not support broadcast yet");
474476

475477
dbl::comm::reorder_to_bf16_for_mix_prec(result);
476478
dbl::comm::reorder_to_bf16_for_mix_prec(self);
@@ -538,6 +540,8 @@ at::Tensor& AtenIpexCPUDev::dil_addmm_out(
538540
if (!result.sizes().equals(inferred_size)) {
539541
result.resize_(inferred_size);
540542
}
543+
TORCH_CHECK(self.sizes().equals(inferred_size),
544+
"addmm not support broadcast yet");
541545

542546
dbl::comm::reorder_to_bf16_for_mix_prec(result);
543547
dbl::comm::reorder_to_bf16_for_mix_prec(self);
@@ -605,6 +609,8 @@ at::Tensor& AtenIpexCPUDev::dil_addbmm_out(
605609
if (!result.sizes().equals(inferred_size)) {
606610
result.resize_(inferred_size);
607611
}
612+
TORCH_CHECK(self.sizes().equals(inferred_size),
613+
"addbmm not support broadcast yet");
608614

609615
dbl::comm::reorder_to_bf16_for_mix_prec(result);
610616
dbl::comm::reorder_to_bf16_for_mix_prec(self);

0 commit comments

Comments
 (0)