DocumentationNeuronDB Documentation
Documentation Branch: You are viewing documentation for the main branch (3.0.0-devel). Select a branch to view its documentation:

ROCm Support

Overview

ROCm support enables AMD GPU acceleration for NeuronDB operations.

Configuration

postgresql.conf

shared_preload_libraries = 'neurondb'
neurondb.gpu_enabled = true
neurondb.gpu_backend = 'rocm'
neurondb.gpu_device = 0

GPU Operations

GPU-accelerated operations

-- GPU-accelerated operations
SELECT vector_l2_distance_gpu(embedding, query) AS distance
FROM documents;

Learn More

For detailed documentation on ROCm setup and AMD GPU configuration, visit: ROCm Support Documentation

Related Topics