Deivanayaki-S commented on code in PR #17930:
URL: https://github.com/apache/tvm/pull/17930#discussion_r2082854026
##########
tests/python/relax/test_frontend_from_fx.py:
##########
@@ -874,6 +874,40 @@ def main(
verify_model(Einsum2(), [([5], "float32"), ([4], "float32")], {},
Expected2)
+def test_outer():
+ class Outer(torch.nn.Module):
+ def forward(self, x, y):
+ return torch.outer(x, y)
+
+ @tvm.script.ir_module
+ class expected:
+ @R.function
+ def main(
+ a: R.Tensor((3,), dtype="float32"), b: R.Tensor((4,),
dtype="float32")
+ ) -> R.Tensor((3, 4), dtype="float32"):
+ with R.dataflow():
+ lv: R.Tensor((3, 4), dtype="float32") = R.outer(a, b)
+ gv: R.Tensor((3, 4), dtype="float32") = lv
+ R.output(gv)
+ return gv
+
+ input_infos = [
+ (
+ [
+ 3,
+ ],
+ "float32",
+ ),
+ (
+ [
+ 4,
+ ],
+ "float32",
+ ),
+ ]
Review Comment:
@Hzfengsy Thanks! I've updated the format, and the lint check has passed
successfully :)
--
This is an automated message from the Apache Git Service.
To respond to the message, please log on to GitHub and use the
URL above to go to the specific comment.
To unsubscribe, e-mail: [email protected]
For queries about this service, please contact Infrastructure at:
[email protected]
