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