mirror of https://github.com/llvm/torch-mlir
ae4724763a
To pass test "MatmulStaticBroadcast_basic" in stablehlo:
```python
class MatmulStaticBroadcast(torch.nn.Module):
def __init__(self):
super().__init__()
@export
@annotate_args([
None,
([4, 1, 6, 7], torch.float32, True),
([8, 1, 5, 7, 6], torch.float32, True),
])
def forward(self, lhs, rhs):
return torch.matmul(lhs, rhs)
@register_test_case(module_factory=lambda: MatmulStaticBroadcast())
def MatmulStaticBroadcast_basic(module, tu: TestUtils):
module.forward(tu.rand(4, 1, 6, 7), tu.rand(8, 1, 5, 7, 6))
```
|
||
|---|---|---|
| .. | ||
| Basic.cpp | ||
| CMakeLists.txt | ||
| GatherScatter.cpp | ||
| Linear.cpp | ||
| Pooling.cpp | ||
| PopulatePatterns.h | ||
| Reduction.cpp | ||
| StablehloLegalizeUtils.cpp | ||
| TorchToStablehlo.cpp | ||
| Utils.cpp | ||
| Utils.h | ||
| ViewLike.cpp | ||