issue/614: fix device class in python

5f0f80d6 · zhangyue · cad2d45a · 5f0f80d6
Commit 5f0f80d6 authored Nov 18, 2025 by zhangyue
Show whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

src/infinirt/infinirt.cc src/infinirt/infinirt.cc +1 -1

No files found.
--- a/src/infinirt/infinirt.cc
+++ b/src/infinirt/infinirt.cc
@@ -23,7 +23,7 @@ __C infiniStatus_t infinirtGetAllDeviceCount(int *count_array) {
        return INFINI_STATUS_NULL_POINTER;
    }
    for (size_t i = 0; i < INFINI_DEVICE_TYPE_COUNT; i++) {
-        if (i == INFINI_DEVICE_ILUVATAR || i == INFINI_DEVICE_QY || i == INFINI_DEVICE_KUNLUN || i == INFINI_DEVICE_HYGON) {
+        if (i == INFINI_DEVICE_ILUVATAR || i == INFINI_DEVICE_HYGON) {
            count_array[i] = 0;
            continue;
        }