torch-mlir

History

Arham Khan 574143448b [E2E][ONNX] torch.multinomial (#3404 ) This PR adds a conversion in the TorchOnnxToTorch pass for the ONNX Multinomial operation. It also adds a TorchToLinalg lowering for the `aten.Multinomial` op and does a light refactor of some repeated code that generates random floating point numbers in `TorchToLinalg/Random.cpp`.	2024-07-16 23:09:39 +05:30
..
main.py	[torch-mlir] bump to llvm/llvm-project@9b78ddf3b2 (#3491 )	2024-06-27 19:28:02 -07:00
xfail_sets.py	[E2E][ONNX] torch.multinomial (#3404 )	2024-07-16 23:09:39 +05:30

This PR adds a conversion in the TorchOnnxToTorch pass for the ONNX
Multinomial operation. It also adds a TorchToLinalg lowering for the
`aten.Multinomial` op and does a light refactor of some repeated code
that generates random floating point numbers in
`TorchToLinalg/Random.cpp`.

2024-07-16 23:09:39 +05:30

main.py

[torch-mlir] bump to llvm/llvm-project@9b78ddf3b2 (#3491 )

2024-06-27 19:28:02 -07:00

xfail_sets.py

[E2E][ONNX] torch.multinomial (#3404 )

2024-07-16 23:09:39 +05:30