[BUG] Parser error with nn.dense

[gussmith23]

Parsing a Relay program which uses nn.dense seems to be finicky! I might be doing something wrong here, let me know!

A minimum working example:

import tvm
from tvm import relay
from tvm.relay.testing.mobilenet import get_workload

x = relay.var('x', shape = (3,))
y = relay.var('y', shape = (3,3))
prog = relay.nn.dense(x, y)
function = relay.Function([x, y], prog)
module = tvm.IRModule.from_expr(function)

print(module.astext())

tvm.parser.fromtext(module.astext())

# if we remove the "None" that is causing the error, we get a different error:
tvm.parser.fromtext(
'''
#[version = "0.0.5"]
def @main(%x: Tensor[(3), float32], %y: Tensor[(3, 3), float32]) -> Tensor[(3), float32] {
  nn.dense(%x, %y) /* ty=Tensor[(3), float32] */
}
'''
    )

[jroesch]

It could be a bug, I've fixed a ton of bugs on #6274, I'll try to add a regression there.

[tqchen]

@gussmith23 please check if the problem persists in main

[gussmith23]

Looks like it's gone!

[gussmith23]

I'm getting another error in the Rust bindings now. Will open a different issue.