Commit 5bce981
authored
Arm backend: Avoid tensor copy warning in quant config (#20687)
The scale product is already a tensor, so wrapping it with torch.tensor
emits a PyTorch copy-construction warning. Use it directly and cast it
to float32 instead.
Signed-off-by: Sebastian Larsson <sebastian.larsson@arm.com>1 parent 3bd4748 commit 5bce981
1 file changed
Lines changed: 3 additions & 3 deletions
| Original file line number | Diff line number | Diff line change | |
|---|---|---|---|
| |||
189 | 189 | | |
190 | 190 | | |
191 | 191 | | |
192 | | - | |
193 | | - | |
194 | | - | |
| 192 | + | |
| 193 | + | |
| 194 | + | |
195 | 195 | | |
196 | 196 | | |
197 | 197 | | |
| |||
0 commit comments