Function hip_atexit_handler

cake-core/src/backends/rocm/mod.rs:41–49

atexit handler: terminates process before HIP's broken atexit cleanup runs. Registered after HIP init, so runs before HIP's handler (LIFO order).

()

/// atexit handler: terminates process before HIP's broken atexit cleanup runs.
/// Registered after HIP init, so runs before HIP's handler (LIFO order).
extern "C" fn hip_atexit_handler() {
    if HIP_INITIALIZED.load(Ordering::Relaxed) {
        // Flush stdio then hard-exit to prevent HIP atexit crash
        unsafe {
            libc::fflush(std::ptr::null_mut());
            libc::syscall(libc::SYS_exit_group, 0);
        }
    }
}

use candle_core::{DType, Device, Result, Tensor, TensorId, D};