| 文件 | 最后提交记录 | 最后更新时间 |
|---|---|---|
build: manually update PyTorch version (#3896) This commit sets the PyTorch and TorchVision version to nightly release 2024-12-01. This commit also updates the test checks in test/python/fx_importer/v2.3/auto_functionalized.py. Failing tests are tracked through https://github.com/llvm/torch-mlir/issues/3796. --------- Signed-off-by: Vivek Khandelwal <vivekkhandelwal1424@gmail.com> | 1 年前 | |
Add Windows build workflow in the CI. (#4041) 1. Adds a workflow to trigger windows build in the CI similar to the existing one for Linux to address the build CI request captured in https://github.com/llvm/torch-mlir/issues/3985 2. Only non-python based LIT unit tests are enabled in the workflow due to failures reported in https://github.com/llvm/torch-mlir/issues/4040 3. The CI is only run against torch-nightly since python based tests are disabled. My understanding is that without the python based tests there's no need to run against the stable version too. If that's not the case, will enable stable as well. Tested that both nightly and stable versions work fine: https://github.com/sahas3/torch-mlir/actions/runs/13477124004 | 1 年前 | |
Add ConstantArgument support to fx_import (#4244) This PR will fix the following issue: [fx_importer NotImplementedError: MultiheadAttention layer with NeedWeight = false](https://github.com/llvm/torch-mlir/issues/4158) The following error was raised before this fix: Python Error: NotImplementedError: OutputKind.USER_OUTPUT for <class 'torch.export.graph_signature.ConstantArgument'>: ConstantArgument(name='', value=None) This occurs for an exported MultiheadAttention layer with "NeedWeight = false" which means weights are not going to be returned by the layer. So, the second output attn_output_weights will be None in this case. | 1 年前 | |
[fx] Fix type determination for multi-return ops and static None returns. (#3081) In practice, this was caught by the way that AOT autograd traces convolution_backward. For the unit test, we just repro it with a custom op. | 2 年前 | |
build: manually update PyTorch version (#4112) This commit sets the PyTorch and TorchVision versions to nightly release 2025-04-23. This commit also updates the fx_importer tests with the changes related to symbolic shape ids introduced in https://github.com/pytorch/pytorch/commit/f649ee73ce28fe8e021ec93f69db4eb8f14285f4. This commit disables some checks because of the mismatch in the resultant IR for PyTorch nightly and stable versions. Those checks will be enabled once they are on the same page. --------- Signed-off-by: Vivek Khandelwal <vivekkhandelwal1424@gmail.com> | 1 年前 |
| 文件 | 最后提交记录 | 最后更新时间 |
|---|---|---|
| 1 年前 | ||
| 1 年前 | ||
| 1 年前 | ||
| 2 年前 | ||
| 1 年前 |