mirror of https://github.com/llvm/torch-mlir
c1167853db
This commit fixes an error in the refine types pass of constant allocation ops. The function used to set the dtype, `fillInDtypeGivenDtypeAndDataType`, takes two torch types as arguments, but a torch type and a standard MLIR type were being passed into it. This commit also fixes the way the dtype was calculated in `visitAtenToDtypeOp`. This op was also passing a standard MLIR type as an argument to the `fillInDtypeGivenDtypeAndDataType` function. Moreover, since the op `aten.to.dtype` has the dtype argument as not optional, all that is needed is to match against the int value to extract the dtype. |
||
---|---|---|
.. | ||
arange.py | ||
argmax.py | ||
backprop.py | ||
basic.py | ||
constant_alloc.py | ||
conv.py | ||
elementwise.py | ||
histogram_binning_calibration.py | ||
index_select.py | ||
main.py | ||
matmul.py | ||
mlp.py | ||
nll_loss.py | ||
norm_like.py | ||
quantized_models.py | ||
reduction.py | ||
reshape_like.py | ||
rng.py | ||
scalar.py | ||
slice_like.py | ||
squeeze.py | ||
table_batch_embedding.py | ||
threshold.py | ||
type_conversion.py | ||
type_promotion.py | ||
vision_models.py | ||
xfail_sets.py |