Skip to content

T
tic
Commit d3b8c869 by Tianqi Chen Committed by GitHub
Options

[TEST] Fix java compilation (#279)

parent 52c8db5b
Showing with 2 additions and 6 deletions
jvm/core/src/test/scripts/test_add_cpu.py
import os import os
import tvm import tvm
from tvm.contrib import cc_compiler as cc from tvm.contrib import cc, util
from tvm.contrib import util
def test_add(target_dir): def test_add(target_dir):
n = tvm.var("n") n = tvm.var("n")
...@@ -21,4 +20,3 @@ if __name__ == "__main__": ...@@ -21,4 +20,3 @@ if __name__ == "__main__":
if len(sys.argv) != 2: if len(sys.argv) != 2:
sys.exit(-1) sys.exit(-1)
test_add(sys.argv[1]) test_add(sys.argv[1])
jvm/core/src/test/scripts/test_add_gpu.py
import os import os
import tvm import tvm
from tvm.contrib import cc_compiler as cc from tvm.contrib import cc, util
from tvm.contrib import util
def test_add(target_dir): def test_add(target_dir):
n = tvm.var("n") n = tvm.var("n")
...@@ -27,4 +26,3 @@ if __name__ == "__main__": ...@@ -27,4 +26,3 @@ if __name__ == "__main__":
if len(sys.argv) != 2: if len(sys.argv) != 2:
sys.exit(-1) sys.exit(-1)
test_add(sys.argv[1]) test_add(sys.argv[1])
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