Module tensor_core_optimization

scirs2_optimize::gpu

Module tensor_core_optimization 

Source
Expand description

Tensor Core optimizations for high-performance GPU acceleration

This module leverages NVIDIA Tensor Cores for accelerated matrix operations in optimization algorithms, providing significant speedup for suitable workloads.

Structsยง

AMPManager
Automatic Mixed Precision (AMP) manager for optimization
TensorCoreOptimizationConfig
Tensor Core acceleration configuration
TensorCoreOptimizer
Tensor Core-accelerated matrix operations for optimization