Skip to content

T
tic
Commit bcea8f6f by tqchen
Options

python3 compatibility

parent e011edd1
Showing with 4 additions and 6 deletions
tests/python/test_basic.py
...@@ -9,7 +9,7 @@ def test_make(): ...@@ -9,7 +9,7 @@ def test_make():
x = tvm.const(1) x = tvm.const(1)
y = tvm.make.IntImm('int32', 1) y = tvm.make.IntImm('int32', 1)
z = x + y z = x + y
print tvm.format_str(z) print(tvm.format_str(z))
def test_ir(): def test_ir():
x = tvm.const(1) x = tvm.const(1)
...@@ -17,7 +17,6 @@ def test_ir(): ...@@ -17,7 +17,6 @@ def test_ir():
z = x + y z = x + y
stmt = tvm.make.Evaluate(z) stmt = tvm.make.Evaluate(z)
assert isinstance(stmt, tvm.stmt.Evaluate) assert isinstance(stmt, tvm.stmt.Evaluate)
print tvm.format_str(stmt)
def test_basic(): def test_basic():
a = tvm.Var('a') a = tvm.Var('a')
...@@ -29,10 +28,9 @@ def test_array(): ...@@ -29,10 +28,9 @@ def test_array():
a = tvm.convert([1,2,3]) a = tvm.convert([1,2,3])
def test_stmt(): def test_stmt():
print tvm.make.Provide('a', [1,2,3], [1,2,3]) tvm.make.For(tvm.Var('i'), 0, 1,
print tvm.make.For('a', 0, 1, tvm.stmt.For.Serial, 0,
tvm.stmt.For.Serial, 0, tvm.make.Evaluate(0))
tvm.make.Evaluate(0))
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment