torch-node.getGPUBackend

function getGPUBackend(): GPUBackend

Get the configured GPU backend from environment.

Reads the TORCH_GPU_BACKEND environment variable and returns the corresponding backend type. If not set or invalid, defaults to 'dawn'.

Returns

GPUBackend– The selected GPU backend type

// Returns 'wgpu-native' if TORCH_GPU_BACKEND=wgpu-native
const backend = getGPUBackend();
console.log(`Using GPU backend: ${backend}`);

function getGPUBackend(): GPUBackend

Get the configured GPU backend from environment.

Reads the TORCH_GPU_BACKEND environment variable and returns the corresponding backend type. If not set or invalid, defaults to 'dawn'.

GPUBackend– The selected GPU backend type

// Returns 'wgpu-native' if TORCH_GPU_BACKEND=wgpu-native
const backend = getGPUBackend();
console.log(`Using GPU backend: ${backend}`);