Skip to content

GitLab

Unverified Commit 66dfb435 authored by PanZezhong1725's avatar PanZezhong1725 Committed by GitHub
Browse files

Merge pull request #330 from InfiniTensor/issue-158-update-infinirt 

issue/158/fix: Update infinirt.cc
parents 6dde36db e25f4ca0
Hide whitespace changes
Inline Side-by-side
Showing with 3 additions and 0 deletions
src/infinirt/infinirt.cc
View file @ 66dfb435
...@@ -70,6 +70,9 @@ __C infiniStatus_t infinirtGetDevice(infiniDevice_t *device_ptr, int *device_id_ ...@@ -70,6 +70,9 @@ __C infiniStatus_t infinirtGetDevice(infiniDevice_t *device_ptr, int *device_id_
case INFINI_DEVICE_KUNLUN: \ case INFINI_DEVICE_KUNLUN: \
_status = infinirt::kunlun::API PARAMS; \ _status = infinirt::kunlun::API PARAMS; \
break; \ break; \
case INFINI_DEVICE_ILUVATAR: \
_status = infinirt::cuda::API PARAMS; \
break; \
default: \ default: \
_status = INFINI_STATUS_DEVICE_TYPE_NOT_SUPPORTED; \ _status = INFINI_STATUS_DEVICE_TYPE_NOT_SUPPORTED; \
} \ } \
......
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