heel 0.1.1

Cross-platform native sandboxing library for running untrusted code
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64

#!/usr/bin/env python3
"""Test GPU access via Metal (macOS)"""

import sys

def test_metal():
    """Test Metal GPU access"""
    try:
        import Metal
        devices = Metal.MTLCopyAllDevices()
        if devices:
            print(f"Found {len(devices)} Metal device(s):")
            for i, device in enumerate(devices):
                print(f"  [{i}] {device.name()}")
            return True
        else:
            print("No Metal devices found")
            return False
    except ImportError:
        print("Metal module not available (pyobjc-framework-Metal not installed)")
        return None
    except Exception as e:
        print(f"Metal access denied or failed: {e}")
        return False

def test_opencl():
    """Test OpenCL GPU access"""
    try:
        import pyopencl as cl
        platforms = cl.get_platforms()
        if platforms:
            print(f"Found {len(platforms)} OpenCL platform(s):")
            for platform in platforms:
                print(f"  Platform: {platform.name}")
                devices = platform.get_devices()
                for device in devices:
                    print(f"    Device: {device.name}")
            return True
        else:
            print("No OpenCL platforms found")
            return False
    except ImportError:
        print("pyopencl not installed")
        return None
    except Exception as e:
        print(f"OpenCL access denied or failed: {e}")
        return False

if __name__ == "__main__":
    print("=== GPU Access Test ===\n")

    print("Testing Metal...")
    metal_result = test_metal()

    print("\nTesting OpenCL...")
    opencl_result = test_opencl()

    print("\n=== Results ===")
    print(f"Metal: {'OK' if metal_result else 'DENIED' if metal_result is False else 'N/A'}")
    print(f"OpenCL: {'OK' if opencl_result else 'DENIED' if opencl_result is False else 'N/A'}")

    # Exit with error if access was denied (not just unavailable)
    if metal_result is False or opencl_result is False:
        sys.exit(1)