Skip to content

Commit 0471ed7

Browse files
authored
Update diopi_functions.yaml
1 parent cc41a20 commit 0471ed7

File tree

1 file changed

+4
-4
lines changed

1 file changed

+4
-4
lines changed

dipu/scripts/autogen_diopi_wrapper/diopi_functions.yaml

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -159,23 +159,23 @@
159159
interface: diopiMulScalar(ctx, out, self, other)
160160

161161
- schema: "mul_.Tensor(Tensor(a!) self, Tensor other) -> Tensor(a!)"
162-
device: [cuda]
162+
device: [cuda, muxi]
163163
interface: diopiMulInp(ctx, self, other)
164164

165165
- schema: "mul_.Tensor(Tensor(a!) self, Tensor other) -> Tensor(a!)"
166-
device: [-cuda, all]
166+
device: [-cuda, -muxi, all]
167167
custom_code_at_the_beginning: |
168168
if (is_scalar_on_cpu(other)) {
169169
return dipu_mul__scalar(self, other.item());
170170
}
171171
interface: diopiMulInp(ctx, self, other)
172172

173173
- schema: "mul.out(Tensor self, Tensor other, *, Tensor(a!) out) -> Tensor(a!)"
174-
device: [cuda]
174+
device: [cuda, muxi]
175175
interface: diopiMul(ctx, out, self, other)
176176

177177
- schema: "mul.out(Tensor self, Tensor other, *, Tensor(a!) out) -> Tensor(a!)"
178-
device: [-cuda, all]
178+
device: [-cuda, -muxi, all]
179179
custom_code_at_the_beginning: |
180180
// if (is_scalar_on_cpu(other)) {
181181
// Pytorch 2.0 has a bug, causing for_each mul passing a cpu scalar tensor. Fixed in PyTorch 2.1

0 commit comments

Comments
 (0)