Added CLDevice.getPartitionType()

ochafik · ochafik · commit caeb6077c3ce · 2015-03-20T01:03:47.000Z
diff --git a/Core/src/main/velocity/com/nativelibs4java/opencl/CLDevice.java b/Core/src/main/velocity/com/nativelibs4java/opencl/CLDevice.java
@@ -734,12 +734,25 @@ public static PartitionType getEnum(long v) {
 
     @InfoName("CL_DEVICE_PARTITION_PROPERTIES")
     public EnumSet<PartitionType> getPartitionProperties() {
+        return getPartitionTypeInfo(CL_DEVICE_PARTITION_PROPERTIES);
+    }
+
+    @InfoName("CL_DEVICE_PARTITION_TYPE")
+    public EnumSet<PartitionType> getPartitionType() {
+        return getPartitionTypeInfo(CL_DEVICE_PARTITION_TYPE);
+    }
+
+    private EnumSet<PartitionType> getPartitionTypeInfo(int info) {
         EnumSet<PartitionType> ret = EnumSet.noneOf(PartitionType.class);
-        Pointer<?> ptr = infos.getPointer(getEntity(), CL_DEVICE_PARTITION_PROPERTIES);;
+        Pointer<?> ptr = infos.getPointer(getEntity(), info);
         if (ptr != null) {
             Pointer<SizeT> props = ptr.as(SizeT.class);
             for (long i = 0, n = props.getValidElements(); i < n; i++) {
-                ret.add(PartitionType.getEnum(props.getSizeTAtIndex(i)));
+                long value = props.getSizeTAtIndex(i);
+                if (value == 0) {
+                  break;
+                }
+                ret.add(PartitionType.getEnum(value));
             }
         }
         return ret;
@@ -756,7 +769,11 @@ public EnumSet<AffinityDomain> getPartitionAffinityDomains() {
         if (ptr != null) {
             Pointer<SizeT> props = ptr.as(SizeT.class);
             for (long i = 0, n = props.getValidElements(); i < n; i++) {
-                ret.add(AffinityDomain.getEnum(props.getSizeTAtIndex(i)));
+                long value = props.getSizeTAtIndex(i);
+                if (value == 0) {
+                  break;
+                }
+                ret.add(AffinityDomain.getEnum(value));
             }
         }
         return ret;
