mirror of https://github.com/llvm/torch-mlir
5d28549c2c
This is to fold the common pattern from Bert inference like: ``` %111 = torch.prim.NumToTensor.Scalar %110 : !torch.int -> !torch.vtensor<[],si64> %112 = torch.aten.Int.Tensor %111 : !torch.vtensor<[],si64> -> !torch.int ``` |
||
---|---|---|
.. | ||
test | ||
torch_mlir/dialects | ||
torch_mlir_e2e_test | ||
CMakeLists.txt | ||
TorchMLIRModule.cpp |