Mila
Deep Neural Network Library
Loading...
Searching...
No Matches
RocmTensorTraits.ixx File Reference

Vulkan-specific tensor trait specializations for explicit graphics compute. More...

#include <cstdint>
#include <string_view>
#include <type_traits>
import Compute.DeviceType;
import Dnn.TensorDataTypeMap;
import Dnn.TensorDataType;

Classes

class  Mila::Dnn::VulkanTensorTraits
 Vulkan-specific traits for abstract tensor data types. More...

Namespaces

namespace  Mila
 Mila main API namespace.

Detailed Description

Vulkan-specific tensor trait specializations for explicit graphics compute.

This module provides Vulkan-specific tensor characteristics and native type mappings for explicit graphics and compute operations. Supports Vulkan's memory model with fine-grained control over device memory allocation, synchronization, and compute pipeline integration across different GPU vendors.