torch.DeviceCapabilities
export interface DeviceCapabilities {
/** Whether device supports compute shaders */
readonly compute: boolean;
/** Whether device supports storage buffers */
readonly storage: boolean;
/** Maximum buffer size in bytes */
readonly maxBufferSize: number;
/** Maximum workgroup size [x, y, z] */
readonly maxWorkgroupSize: readonly [number, number, number];
/** Maximum workgroups per dimension */
readonly maxWorkgroupsPerDimension: number;
/** Supported data types */
readonly supportedDTypes: readonly string[];
}- readonly
compute(boolean) - – Whether device supports compute shaders
- readonly
storage(boolean) - – Whether device supports storage buffers
- readonly
maxBufferSize(number) - – Maximum buffer size in bytes
- readonly
maxWorkgroupSize(readonly [number, number, number]) - – Maximum workgroup size [x, y, z]
- readonly
maxWorkgroupsPerDimension(number) - – Maximum workgroups per dimension
- readonly
supportedDTypes(readonly string[]) - – Supported data types
Device capabilities.