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

issue/158/fix: Update infinirt.cc

Merge pull request #330 from InfiniTensor/issue-158-update-infinirt
issue/158/fix: Update infinirt.cc
66dfb435 · PanZezhong1725 · GitHub · 6dde36db · e25f4ca0 · 66dfb435
Unverified Commit 66dfb435 authored Jul 16, 2025 by PanZezhong1725 Committed by GitHub Jul 16, 2025
Hide whitespace changes
Inline Side-by-side

Showing with 3 additions and 0 deletions

src/infinirt/infinirt.cc src/infinirt/infinirt.cc +3 -0

No files found.
--- a/src/infinirt/infinirt.cc
+++ b/src/infinirt/infinirt.cc
@@ -70,6 +70,9 @@ __C infiniStatus_t infinirtGetDevice(infiniDevice_t *device_ptr, int *device_id_
        case INFINI_DEVICE_KUNLUN:                                     \
            _status = infinirt::kunlun::API PARAMS;                    \
            break;                                                     \
+        case INFINI_DEVICE_ILUVATAR:                                   \
+            _status = infinirt::cuda::API PARAMS;                      \
+            break;                                                     \
        default:                                                       \
            _status = INFINI_STATUS_DEVICE_TYPE_NOT_SUPPORTED;         \
        }                                                              \