Struct RocmBackend 

pub struct RocmBackend { /* private fields */ }

ROCm backend using hipTENSOR via runtime dlopen.

Status: Not yet implemented. All methods currently return errors. The type exists to define the intended API surface. plan() and execute() return Err(DeviceError). load_hiptensor() on crate::BackendRegistry also returns an error.

When implemented, will be loaded at runtime from a user-provided .so path with no compile-time ROCm SDK dependency. Will implement the semiring-family traits for standard arithmetic on AMD GPUs.

hipTENSOR natively supports contraction, reduction, and elementwise building blocks. Structural permute stays a tensor view, and any required materialization path is modeled through MakeContiguous. AntiTrace and AntiDiag will be composed via Contract(eye, dC).

Examples

// Aspirational API — not yet functional.
use tenferro_prims::{RocmBackend, BackendRegistry};

let mut registry = BackendRegistry::new();
registry.load_hiptensor("/usr/lib/libhiptensor.so").unwrap();

Implementations

impl RocmBackend

pub fn resolve_conj<T: Scalar + Conjugate>( _ctx: &mut RocmContext, src: &Tensor<T>, ) -> Tensor<T>

Materialize a lazily-conjugated tensor on GPU.

Status: Stub fallback. If data is CPU-accessible, this materializes conjugation into a new non-conjugated tensor. Otherwise it returns a clone of src.

When implemented, will use the analytic/scalar family execution traits to produce a new tensor with conjugated = false.

Trait Implementations

impl Drop for RocmBackend

fn drop(&mut self)

Executes the destructor for this type.

impl<S: Scalar> TensorAnalyticPrims<Standard<S>> for RocmBackend

type Plan = ()

type Context = RocmContext

fn plan( _ctx: &mut Self::Context, desc: &AnalyticPrimsDescriptor, _shapes: &[&[usize]], ) -> Result<Self::Plan>

Plan an analytic-family operation for the given input/output shapes.

fn execute( _ctx: &mut Self::Context, _plan: &Self::Plan, _alpha: S, _inputs: &[&Tensor<S>], _beta: S, _output: &mut Tensor<S>, ) -> Result<()>

Execute a previously planned analytic-family operation.

fn has_analytic_support(_desc: AnalyticPrimsDescriptor) -> bool

Report whether the backend advertises support for the given descriptor.

impl<Input> TensorComplexRealPrims<Input> for RocmBackend

where Input: ComplexFloat + Scalar, Input::Real: Scalar,

type Real = <Input as ComplexFloat>::Real

The real-valued output scalar type.

type Plan = ()

Backend plan type.

type Context = RocmContext

Backend execution context.

fn plan( _ctx: &mut Self::Context, desc: &ComplexRealPrimsDescriptor, _shapes: &[&[usize]], ) -> Result<Self::Plan>

Plan a complex-to-real unary operation for the given input/output shapes.

fn execute( _ctx: &mut Self::Context, _plan: &Self::Plan, _alpha: Input::Real, _inputs: &[&Tensor<Input>], _beta: Input::Real, _output: &mut Tensor<Self::Real>, ) -> Result<()>

Execute a previously planned complex-to-real unary operation.

fn has_complex_real_support(_desc: ComplexRealPrimsDescriptor) -> bool

Report whether the backend advertises support for the given descriptor.

impl<Input> TensorComplexScalePrims<Input> for RocmBackend

where Input: ComplexFloat + Scalar, Input::Real: Scalar + Send + Sync,

type Plan = ()

Backend plan type.

type Context = RocmContext

Backend execution context.

fn plan( _ctx: &mut Self::Context, desc: &ComplexScalePrimsDescriptor, _shapes: &[&[usize]], ) -> Result<Self::Plan>

Plan a complex-by-real pointwise operation for the given shapes.

fn execute( _ctx: &mut Self::Context, _plan: &Self::Plan, _alpha: Input, _lhs: &Tensor<Input>, _rhs: &Tensor<Input::Real>, _beta: Input, _output: &mut Tensor<Input>, ) -> Result<()>

Execute a previously planned complex-by-real pointwise operation.

fn has_complex_scale_support(_desc: ComplexScalePrimsDescriptor) -> bool

Report whether the backend advertises support for the given descriptor.

impl<S: Scalar> TensorIndexingPrims<Standard<S>> for RocmBackend

type Plan = ()

Backend-specific execution plan.

type Context = RocmContext

Backend execution context.

fn plan( _ctx: &mut Self::Context, desc: &IndexingPrimsDescriptor, _shapes: &[&[usize]], ) -> Result<Self::Plan>

Plan an indexing operation for the given input/index/output shapes.

fn execute( _ctx: &mut Self::Context, _plan: &Self::Plan, _inputs: &[&Tensor<S>], _indices: &Tensor<i64>, _output: &mut Tensor<S>, ) -> Result<()>

Execute a previously planned indexing operation.

fn has_indexing_support(_desc: IndexingPrimsDescriptor) -> bool

Report whether the backend advertises support for the given descriptor.

impl<S: Scalar + NumCast> TensorMetadataCastPrims<S> for RocmBackend

type Plan = MetadataCastPrimsDescriptor

Backend plan type.

type Context = RocmContext

Backend execution context.

fn plan( _ctx: &mut RocmContext, desc: &MetadataCastPrimsDescriptor, _shapes: &[&[usize]], ) -> Result<Self::Plan>

Plan a metadata-to-scalar bridge operation.

fn execute( _ctx: &mut RocmContext, _plan: &Self::Plan, _alpha: S, _inputs: &[MetadataScalarTensorRef<'_, S>], _beta: S, _output: &mut Tensor<S>, ) -> Result<()>

Execute a previously planned metadata-to-scalar bridge operation.

fn has_metadata_cast_support(_desc: MetadataCastPrimsDescriptor) -> bool

Report whether the backend advertises support for the given descriptor.

impl TensorMetadataPrims for RocmBackend

type Plan = MetadataPrimsDescriptor

Backend plan type.

type Context = RocmContext

Backend execution context.

fn plan( _ctx: &mut RocmContext, desc: &MetadataPrimsDescriptor, _inputs: &[MetadataTensorRef<'_>], _output: MetadataTensorMut<'_>, ) -> Result<Self::Plan>

Plan a metadata-family operation for the given input and output tensor handles.

fn execute( _ctx: &mut RocmContext, _plan: &Self::Plan, _inputs: &[MetadataTensorRef<'_>], _output: MetadataTensorMut<'_>, ) -> Result<()>

Execute a previously planned metadata-family operation in overwrite mode.

fn has_metadata_support(_desc: MetadataPrimsDescriptor) -> bool

Report whether the backend advertises support for the given descriptor.

impl TensorRngPrims<Standard<f64>> for RocmBackend

type Plan = (RngPrimsDescriptor, Vec<usize>)

Backend-specific execution plan.

type Context = RocmContext

Backend execution context.

fn plan( _ctx: &mut Self::Context, desc: &RngPrimsDescriptor, _shapes: &[&[usize]], ) -> Result<Self::Plan>

Plan a tensor RNG operation for the given output shape.

fn execute( _ctx: &mut Self::Context, _plan: &Self::Plan, _generator: &mut Generator, _output: &mut Tensor<f64>, ) -> Result<()>

Execute a previously planned RNG operation.

fn has_rng_support(_desc: RngPrimsDescriptor) -> bool

Report whether the backend advertises support for the given descriptor.

impl TensorRngPrims<Standard<i32>> for RocmBackend

type Plan = (RngPrimsDescriptor, Vec<usize>)

Backend-specific execution plan.

type Context = RocmContext

Backend execution context.

fn plan( _ctx: &mut Self::Context, desc: &RngPrimsDescriptor, _shapes: &[&[usize]], ) -> Result<Self::Plan>

Plan a tensor RNG operation for the given output shape.

fn execute( _ctx: &mut Self::Context, _plan: &Self::Plan, _generator: &mut Generator, _output: &mut Tensor<i32>, ) -> Result<()>

Execute a previously planned RNG operation.

fn has_rng_support(_desc: RngPrimsDescriptor) -> bool

Report whether the backend advertises support for the given descriptor.

impl<S: Scalar> TensorScalarPrims<Standard<S>> for RocmBackend

type Plan = ()

type Context = RocmContext

fn plan( _ctx: &mut Self::Context, desc: &ScalarPrimsDescriptor, _shapes: &[&[usize]], ) -> Result<Self::Plan>

Plan a scalar-family operation for the given input/output shapes.

fn execute( _ctx: &mut Self::Context, _plan: &Self::Plan, _alpha: S, _inputs: &[&Tensor<S>], _beta: S, _output: &mut Tensor<S>, ) -> Result<()>

Execute a previously planned scalar-family operation.

fn has_scalar_support(_desc: ScalarPrimsDescriptor) -> bool

Report whether the backend advertises support for the given descriptor.

impl<S: Scalar> TensorSemiringCore<Standard<S>> for RocmBackend

type Plan = RocmPlan<S>

Backend-specific plan type.

type Context = RocmContext

Backend-specific execution context.

fn plan( _ctx: &mut RocmContext, _desc: &SemiringCoreDescriptor, _shapes: &[&[usize]], ) -> Result<RocmPlan<S>>

Plan a semiring-core operation.

fn execute( _ctx: &mut RocmContext, _plan: &RocmPlan<S>, _alpha: S, _inputs: &[&Tensor<S>], _beta: S, _output: &mut Tensor<S>, ) -> Result<()>

Execute a semiring-core operation.

impl<S: Scalar> TensorSemiringFastPath<Standard<S>> for RocmBackend

type Plan = RocmPlan<S>

Backend-specific plan type.

type Context = RocmContext

Backend-specific execution context.

fn plan( _ctx: &mut RocmContext, _desc: &SemiringFastPathDescriptor, _shapes: &[&[usize]], ) -> Result<RocmPlan<S>>

Plan an optional semiring fast path.

fn execute( _ctx: &mut RocmContext, _plan: &RocmPlan<S>, _alpha: S, _inputs: &[&Tensor<S>], _beta: S, _output: &mut Tensor<S>, ) -> Result<()>

Execute an optional semiring fast path.

fn has_fast_path(_desc: SemiringFastPathDescriptor) -> bool

Query whether the optional path is available.

impl<S: Scalar + PartialOrd> TensorSortPrims<Standard<S>> for RocmBackend

type Plan = ()

Backend-specific execution plan.

type Context = RocmContext

Backend execution context.

fn plan( _ctx: &mut Self::Context, desc: &SortPrimsDescriptor, _shapes: &[&[usize]], ) -> Result<Self::Plan>

Plan a sort operation for the given input shape.

fn execute( _ctx: &mut Self::Context, _plan: &Self::Plan, _input: &Tensor<S>, _values_out: &mut Tensor<S>, _indices_out: &mut Tensor<i64>, ) -> Result<()>

Execute a previously planned sort operation.

fn has_sort_support(_desc: &SortPrimsDescriptor) -> bool

Report whether the backend advertises support for the given descriptor.

impl Send for RocmBackend

Safety

RocmBackend can be safely sent across threads because:

• The _handle is an opaque pointer to a hipTENSOR handle
• The _lib (libloading::Library) is thread-safe after loading
• The handle is read-only after construction
• Drop clears the handle before the library is unloaded, preventing use-after-free

impl Sync for RocmBackend

Safety

RocmBackend can be safely shared across threads because:

• The hipTENSOR handle is designed for concurrent use from multiple threads
• The library handle (_lib) is read-only after construction
• Symbol lookup via dlsym is thread-safe on POSIX systems
• Drop uses &mut self, ensuring exclusive access during cleanup