We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 8e8b247 commit d5b42c2Copy full SHA for d5b42c2
csrc/ops/alias.cpp
@@ -8,7 +8,6 @@
8
#include <expr_simplifier.h>
9
#include <ir/builder.h>
10
#include <ir/utils.h>
11
-#include <multidevice/utils.h>
12
#include <ops/alias.h>
13
#include <ops/arith.h>
14
#include <ops/utils.h>
@@ -948,12 +947,8 @@ TensorView* broadcast(
948
947
.iter_type(IterType::Broadcast)
949
.build());
950
} 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);
+ out_domain.push_back(
+ IterDomainBuilder(inp_domain[iinp]).resetSchedulingParams().build());
957
iinp++;
958
}
959
ibdim++;
0 commit comments