Skip to content

Commit d5b42c2

Browse files
committed
remove now unnecessary patch on broadcast
1 parent 8e8b247 commit d5b42c2

File tree

1 file changed

+2
-7
lines changed

1 file changed

+2
-7
lines changed

csrc/ops/alias.cpp

+2-7
Original file line numberDiff line numberDiff line change
@@ -8,7 +8,6 @@
88
#include <expr_simplifier.h>
99
#include <ir/builder.h>
1010
#include <ir/utils.h>
11-
#include <multidevice/utils.h>
1211
#include <ops/alias.h>
1312
#include <ops/arith.h>
1413
#include <ops/utils.h>
@@ -948,12 +947,8 @@ TensorView* broadcast(
948947
.iter_type(IterType::Broadcast)
949948
.build());
950949
} else {
951-
auto inp_id = inp_domain[iinp];
952-
auto out_id = IterDomainBuilder(inp_id).resetSchedulingParams().build();
953-
if (inp_id->isDeviceDim()) {
954-
out_id->parallelize(inp_id->getParallelType());
955-
}
956-
out_domain.push_back(out_id);
950+
out_domain.push_back(
951+
IterDomainBuilder(inp_domain[iinp]).resetSchedulingParams().build());
957952
iinp++;
958953
}
959954
ibdim++;

0 commit comments

Comments
 (0)