Skip to content

GitLab

Unverified Commit 359c0db9 authored by PanZezhong1725's avatar PanZezhong1725 Committed by GitHub
Browse files

Merge pull request #20 from InfiniTensor/iluvatar 

support iluvatar device type
parents f59c7bf5 cecc9b5d
Hide whitespace changes
Inline Side-by-side
Showing with 3 additions and 0 deletions
scripts/jiuge.py
View file @ 359c0db9
......@@ -602,6 +602,8 @@ def test():
device_type = DeviceType.DEVICE_TYPE_METAX
elif sys.argv[1] == "--moore":
device_type = DeviceType.DEVICE_TYPE_MOORE
elif sys.argv[1] == "--iluvatar":
device_type = DeviceType.DEVICE_TYPE_ILUVATAR
else:
print(
"Usage: python jiuge.py [--cpu | --nvidia| --cambricon | --ascend | --metax | --moore] <path/to/model_dir> [n_device]"
......
scripts/libinfinicore_infer.py
View file @ 359c0db9
......@@ -33,6 +33,7 @@ class DeviceType(ctypes.c_int):
DEVICE_TYPE_ASCEND = 3
DEVICE_TYPE_METAX = 4
DEVICE_TYPE_MOORE = 5
DEVICE_TYPE_ILUVATAR = 6
class JiugeMetaCStruct(ctypes.Structure):
......
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