add comments for test skip condition.

dmlc · Jul 18, 2024 · 2dc96d9 · 2dc96d9
1 parent 537b815
commit 2dc96d9
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/tests/python/pytorch/graphbolt/impl/test_gpu_cached_feature.py b/tests/python/pytorch/graphbolt/impl/test_gpu_cached_feature.py
@@ -150,10 +150,12 @@ def test_gpu_cached_feature_read_async(dtype, pin_memory):
             values = next(reader)
         assert torch.equal(values.wait(), a_cuda[ids])
 
-    # pin_memory is not applicable to DiskBasedFeature.
     if (
+        # pin_memory is not applicable to DiskBasedFeature.
         pin_memory
+        # DiskBasedFeature's only backend is io_uring.
         or not torch.ops.graphbolt.detect_io_uring()
+        # numpy.save does not work with bfloat16.
         or dtype == torch.bfloat16
     ):
         return