Commit 953fa0e
authored
Do not get numel for unbacked symint (#12490)
Summary:
The current code in _check_inputs_are_valid_dtypes assumes that the
tensor does not have any dim that is unbacked symint.
Factorized joiner contains a data dependent intermediate tensor, where
the batch dim is unbacked: N7582435. Currently the code fails there.
This diff only checks for the numel() if the tensor dimensions are not
unbacked.
Differential Revision: D783438751 parent 1f885b9 commit 953fa0e
1 file changed
+5
-2
lines changed| Original file line number | Diff line number | Diff line change | |
|---|---|---|---|
| |||
16 | 16 | | |
17 | 17 | | |
18 | 18 | | |
| 19 | + | |
19 | 20 | | |
20 | 21 | | |
21 | 22 | | |
| |||
168 | 169 | | |
169 | 170 | | |
170 | 171 | | |
171 | | - | |
172 | | - | |
| 172 | + | |
| 173 | + | |
| 174 | + | |
| 175 | + | |
173 | 176 | | |
174 | 177 | | |
175 | 178 | | |
| |||
0 commit comments