File tree Expand file tree Collapse file tree 1 file changed +6
-0
lines changed Expand file tree Collapse file tree 1 file changed +6
-0
lines changed Original file line number Diff line number Diff line change @@ -471,6 +471,8 @@ at::Tensor& AtenIpexCPUDev::dil_baddbmm_out(
471
471
if (!result.sizes ().equals (inferred_size)) {
472
472
result.resize_ (inferred_size);
473
473
}
474
+ TORCH_CHECK (self.sizes ().equals (inferred_size),
475
+ " baddbmm not support broadcast yet" );
474
476
475
477
dbl::comm::reorder_to_bf16_for_mix_prec (result);
476
478
dbl::comm::reorder_to_bf16_for_mix_prec (self);
@@ -538,6 +540,8 @@ at::Tensor& AtenIpexCPUDev::dil_addmm_out(
538
540
if (!result.sizes ().equals (inferred_size)) {
539
541
result.resize_ (inferred_size);
540
542
}
543
+ TORCH_CHECK (self.sizes ().equals (inferred_size),
544
+ " addmm not support broadcast yet" );
541
545
542
546
dbl::comm::reorder_to_bf16_for_mix_prec (result);
543
547
dbl::comm::reorder_to_bf16_for_mix_prec (self);
@@ -605,6 +609,8 @@ at::Tensor& AtenIpexCPUDev::dil_addbmm_out(
605
609
if (!result.sizes ().equals (inferred_size)) {
606
610
result.resize_ (inferred_size);
607
611
}
612
+ TORCH_CHECK (self.sizes ().equals (inferred_size),
613
+ " addbmm not support broadcast yet" );
608
614
609
615
dbl::comm::reorder_to_bf16_for_mix_prec (result);
610
616
dbl::comm::reorder_to_bf16_for_mix_prec (self);
You can’t perform that action at this time.
0 commit comments