tokitai-operator 0.1.0

//! Bridge module: re-exposes the ROCm/HIP AdamW kernel.
//!
//! The original `src/backend/hip_adamw.rs` uses `use
//! tokitai_operator::...` paths which only resolve when the file
//! is part of the library. This module re-exposes the kernel
//! from the bin/ context, so the training driver can call it
//! without re-implementing the wrapper.
//!
// Bridge module that re-exposes the ROCm/HIP AdamW kernel wrapper from
// library code. The original `src/backend/hip_adamw.rs` uses
// `use tokitai_operator::...` paths which only resolve when the file is
// `include!`d from an integration test (the test crate is named
// `tokitai_operator`). We need the kernel callable from the library,
// so we re-alias `crate` as `tokitai_operator` for the scope of the
// `include!`.
//
// All public items from `hip_adamw.rs` (constants, the kernel source
// fingerprint, the CPU oracle, fp16 conversion helpers, and the two
// GPU runners) become items of this module.

pub use crate as tokitai_operator;

// The included file imports `std::path::PathBuf` (used in its
// test-context compilation but unused here in the library context).
// Wrap the include in an inner module so the `allow` only applies to
// the bridge and we keep the include as a top-level textual copy of
// the kernel contract.
#[allow(unused_imports)]
mod _hip_adamw_bridge {
    pub use crate as tokitai_operator;
    include!("../backend/hip_adamw.rs");
}
pub use _hip_adamw_bridge::*;