apache / tvm

Open deep learning compiler stack for cpu, gpu and specialized accelerators

Home Page:https://tvm.apache.org/

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

[Bug] relay parser not support var name contain "/"

sleepcoo opened this issue · comments

def test_simple():
    code = """
#[version = "0.0.5"]
def @simple[A, B](%fdd/ddsa: fn(A) -> B, %xs: A) -> B {
  %fdd/ddsa(%xs)
}
def @main(%l: Tensor[(5, 5), float32]) -> Tensor[(5, 5), float32] {
  %0 = fn[A](%x: A) -> A {
    %x
  };
  @simple(%0, %l)
}
"""
    # error
    mod = tvm.parser.fromtext(code)

Your simple code works fine with the latest TVM. Any other error information?

it would be good if you can try out Relax