support string option when create cuda/rocm/... target (#1071)

python/tvm/target.py

@@ -349,10 +349,10 @@ def cuda(options=None):
 
     Parameters
     ----------
-    options : list of str
+    options : str or list of str
         Additional options
     """
-    options = options if options else []
+    options = _merge_opts([], options)
     return _api_internal._TargetCreate("cuda", *options)
 
 
@@ -361,10 +361,10 @@ def rocm(options=None):
 
     Parameters
     ----------
-    options : list of str
+    options : str or list of str
         Additional options
     """
-    options = options if options else []
+    options = _merge_opts([], options)
     return _api_internal._TargetCreate("rocm", *options)
 
 
@@ -373,7 +373,7 @@ def rasp(options=None):
 
     Parameters
     ----------
-    options : list of str
+    options : str or list of str
         Additional options
     """
     opts = ["-device=rasp",
@@ -389,7 +389,7 @@ def mali(options=None):
 
     Parameters
     ----------
-    options : list of str
+    options : str or list of str
         Additional options
     """
     opts = ["-device=mali"]
@@ -402,10 +402,10 @@ def opengl(options=None):
 
     Parameters
     ----------
-    options : list of str
+    options : str or list of str
         Additional options
     """
-    options = options if options else []
+    options = _merge_opts([], options)
     return _api_internal._TargetCreate("opengl", *options)
 
 

tests/python/unittest/test_lang_target.py

@@ -43,6 +43,7 @@ def test_target_string_parse():
     assert target.options == ['-libs=cublas,cudnn']
     assert target.keys == ['cuda', 'gpu']
     assert target.libs == ['cublas', 'cudnn']
+    assert str(target) == str(tvm.target.cuda("-libs=cublas,cudnn"))
 
 if __name__ == "__main__":
     test_target_dispatch()