Mila::Dnn::PrecisionSupportedOnDevice Concept Reference

Concept to validate precision is supported on a device at compile-time. More...

Concept definition

    template<TensorDataType TPrecision, Compute::DeviceType TDeviceType>
    concept PrecisionSupportedOnDevice =
        ValidFloatTensorDataType<TPrecision> &&
            ((TDeviceType == Compute::DeviceType::Cpu && TensorDataTypeTraits<TPrecision>::supported_on_cpu) ||
            (TDeviceType == Compute::DeviceType::Cuda && TensorDataTypeTraits<TPrecision>::supported_on_cuda) ||
            (TDeviceType == Compute::DeviceType::Metal && TensorDataTypeTraits<TPrecision>::supported_on_metal))

Detailed Description

Concept to validate precision is supported on a device at compile-time.