Here is a list of all namespace functions with links to the namespace documentation for each function:

- a -

• activationTypeToString() : Mila::Dnn
• adamw_update() : Mila::Dnn::Compute
• add() : Mila::Dnn
• ApproximationMethodToString() : Mila::Dnn
• archiveModeToString() : Mila::Dnn::Serialization
• attentionTypeToString() : Mila::Dnn

- b -

• build_att_value_decode_plan() : Mila::Dnn::Compute::Cuda::Gqa::Detail, Mila::Dnn::Compute::Cuda::MultiHeadAttention::Detail
• build_att_value_decode_plan_optimized() : Mila::Dnn::Compute::Cuda::Gqa::Detail
• build_att_value_plan() : Mila::Dnn::Compute::Cuda::Gqa::Detail, Mila::Dnn::Compute::Cuda::MultiHeadAttention::Detail
• build_att_value_prefill_plan() : Mila::Dnn::Compute::Cuda::Gqa::Detail
• build_att_value_prefill_plan_optimized() : Mila::Dnn::Compute::Cuda::Gqa::Detail
• build_backward_att_plan() : Mila::Dnn::Compute::Cuda::Gqa::Detail, Mila::Dnn::Compute::Cuda::MultiHeadAttention::Detail
• build_backward_input_plan() : Mila::Dnn::Compute::Cuda::Linear::Detail
• build_backward_k_plan() : Mila::Dnn::Compute::Cuda::Gqa::Detail, Mila::Dnn::Compute::Cuda::MultiHeadAttention::Detail
• build_backward_q_plan() : Mila::Dnn::Compute::Cuda::Gqa::Detail, Mila::Dnn::Compute::Cuda::MultiHeadAttention::Detail
• build_backward_v_plan() : Mila::Dnn::Compute::Cuda::Gqa::Detail, Mila::Dnn::Compute::Cuda::MultiHeadAttention::Detail
• build_backward_weight_plan() : Mila::Dnn::Compute::Cuda::Linear::Detail
• build_forward_plan() : Mila::Dnn::Compute::Cuda::Linear::Detail
• build_linear_plan() : Mila::Dnn::Compute::Cuda
• build_plan() : Mila::Dnn::Compute::Cuda
• build_qk_decode_plan() : Mila::Dnn::Compute::Cuda::Gqa::Detail, Mila::Dnn::Compute::Cuda::MultiHeadAttention::Detail
• build_qk_decode_plan_optimized() : Mila::Dnn::Compute::Cuda::Gqa::Detail
• build_qk_prefill_plan() : Mila::Dnn::Compute::Cuda::Gqa::Detail
• build_qk_prefill_plan_optimized() : Mila::Dnn::Compute::Cuda::Gqa::Detail
• build_qk_score_plan() : Mila::Dnn::Compute::Cuda::Gqa::Detail, Mila::Dnn::Compute::Cuda::MultiHeadAttention::Detail
• build_strided_plan() : Mila::Dnn::Compute::Cuda

- c -

• cast_context_() : Mila::Dnn::Compute
• ceil_div() : Mila::Dnn::Compute::Cuda
• checkDevice() : Mila::Dnn::Compute::Cuda
• compute_bias_gradient() : Mila::Dnn::Compute::Cuda::Linear::Detail
• computeArchitectureBuckets() : Mila::Dnn::Compute::Cuda
• computeAxisPartition() : Mila::Dnn
• computeNormalizedShapePartition() : Mila::Dnn
• computeNumElements() : Mila::Dnn
• computePrefillChunkSize() : Mila::Dnn
• connectionTypeToString() : Mila::Dnn
• copy() : Mila::Dnn
• copyFromBlob() : Mila::Dnn
• copyFromBlobWithConversion() : Mila::Dnn
• copyHostToHostImpl() : Mila::Dnn::Compute::Cpu::Detail
• copyHostToHostWithConversionImpl() : Mila::Dnn::Compute::Cpu::Detail
• createExecutionContext() : Mila::Dnn::Compute
• cublaslt_compute_types() : Mila::Dnn::Compute::Cuda
• cublaslt_cuda_data_type() : Mila::Dnn::Compute::Cuda
• cublasLtCheckStatus() : Mila::Dnn::Compute
• cuda_mha_forward_fp16() : Mila::Dnn::Compute
• cuda_mha_forward_fp32() : Mila::Dnn::Compute
• cuda_softmax_crossentropy_backward() : Mila::Dnn::Compute
• cuda_softmax_crossentropy_forward() : Mila::Dnn::Compute
• cuda_split3_bf16() : Mila::Dnn::Compute::Cuda
• cuda_split3_fp32() : Mila::Dnn::Compute::Cuda
• cudaCheckLastError() : Mila::Dnn::Compute
• cudaCheckStatus() : Mila::Dnn::Compute

- d -

• debugDumpTensor() : Mila::Dnn
• deviceTypeToString() : Mila::Dnn::Compute
• divide() : Mila::Dnn
• dtypeToTensorDataType() : Mila::Dnn::Serialization
• dump_2d_rowmajor_host() : Mila::Dnn::Compute::Cuda
• dump_tensor() : Mila::Dnn::Compute::Cuda

- e -

• encodingTypeToString() : Mila::Dnn
• execute_linear_plan() : Mila::Dnn::Compute::Cuda
• execute_plan() : Mila::Dnn::Compute::Cuda

- f -

• fill() : Mila::Dnn
• fill_normal() : Mila::Dnn
• fill_uniform() : Mila::Dnn
• findCudaDevice() : Mila::Dnn::Compute::Cuda
• formatBytes() : Mila::Dnn::Detail, Mila::Dnn::detail
• fromString() : Mila::Dnn
• fromTypeId() : Mila::Dnn

- g -

• get_alignment() : Mila::Dnn::Detail
• getAPIVersion() : Mila
• getBestDevice() : Mila::Dnn::Compute
• getBestDeviceId() : Mila::Dnn::Compute::Cuda
• getBucket() : Mila::Dnn::Compute::Cuda
• getDeviceCount() : Mila::Dnn::Compute::Cuda, Mila::Dnn::Compute
• getDriverVersion() : Mila::Dnn::Compute::Cuda
• getRegisteredOperations() : Mila::Dnn::Compute
• getRuntimeVersion() : Mila::Dnn::Compute::Cuda
• getStorageSize() : Mila::Dnn::Detail, Mila::Dnn::detail
• GPT2_Large() : Mila::Dnn
• GPT2_Medium() : Mila::Dnn
• GPT2_Small() : Mila::Dnn
• GPT2_XL() : Mila::Dnn

- i -

• indexToString() : Mila::Dnn
• initialize() : Mila
• isOperationRegistered() : Mila::Dnn::Compute

- l -

• launch_abs_kernel() : Mila::Dnn::Compute::Cuda::Kernels
• launch_array_fill_typed() : Mila::Dnn::Compute::Cuda
• launch_constant_fill_typed() : Mila::Dnn::Compute::Cuda
• launch_convert_copy_kernel() : Mila::Dnn::Compute::Cuda
• launch_elementwise_add_kernel() : Mila::Dnn::Compute::Cuda::Kernels
• launch_elementwise_divide_kernel() : Mila::Dnn::Compute::Cuda::Kernels
• launch_elementwise_equal_kernel() : Mila::Dnn::Compute::Cuda::Kernels
• launch_elementwise_greater_kernel() : Mila::Dnn::Compute::Cuda::Kernels
• launch_elementwise_less_kernel() : Mila::Dnn::Compute::Cuda::Kernels
• launch_elementwise_max_kernel() : Mila::Dnn::Compute::Cuda::Kernels
• launch_elementwise_min_kernel() : Mila::Dnn::Compute::Cuda::Kernels
• launch_elementwise_multiply_kernel() : Mila::Dnn::Compute::Cuda::Kernels
• launch_elementwise_subtract_kernel() : Mila::Dnn::Compute::Cuda::Kernels
• launch_fast_copy_kernel() : Mila::Dnn::Compute::Cuda
• launch_max_reduction_kernel() : Mila::Dnn::Compute::Cuda::Kernels
• launch_mean_reduction_kernel() : Mila::Dnn::Compute::Cuda::Kernels
• launch_min_reduction_kernel() : Mila::Dnn::Compute::Cuda::Kernels
• launch_negate_kernel() : Mila::Dnn::Compute::Cuda::Kernels
• launch_scalar_add_kernel() : Mila::Dnn::Compute::Cuda::Kernels
• launch_scalar_divide_kernel() : Mila::Dnn::Compute::Cuda::Kernels
• launch_scalar_multiply_kernel() : Mila::Dnn::Compute::Cuda::Kernels
• launch_scalar_subtract_kernel() : Mila::Dnn::Compute::Cuda::Kernels
• launch_scale_shift() : Mila::Dnn::Compute::Cuda
• launch_sqrt_kernel() : Mila::Dnn::Compute::Cuda::Kernels
• launch_square_kernel() : Mila::Dnn::Compute::Cuda::Kernels
• launch_sum_reduction_kernel() : Mila::Dnn::Compute::Cuda::Kernels
• listDevicesByName() : Mila::Dnn::Compute
• listDevicesByType() : Mila::Dnn::Compute
• Llama2_13B() : Mila::Dnn
• Llama2_70B() : Mila::Dnn
• Llama2_7B() : Mila::Dnn
• Llama3_1_405B() : Mila::Dnn
• Llama3_1_70B() : Mila::Dnn
• Llama3_1_8B() : Mila::Dnn
• Llama3_2_1B() : Mila::Dnn
• Llama3_2_3B() : Mila::Dnn
• Llama3_70B() : Mila::Dnn
• Llama3_8B() : Mila::Dnn

- m -

• makeOpInstance() : Mila::Dnn::Compute
• multiply() : Mila::Dnn

- n -

• normTypeToString() : Mila::Dnn

- o -

• operationTypeToString() : Mila::Dnn::Compute
• operator*() : Mila::Dnn
• operator+() : Mila::Dnn
• operator-() : Mila::Dnn
• operator/() : Mila::Dnn
• operator<<() : Mila::Dnn

- p -

• parseTensorDataType() : Mila::Dnn
• print_stats() : Mila::Dnn::Compute::Cuda

- q -

• quantize_fp4_per_group() : Mila::Dnn::Compute::Cuda::Linear::Detail
• quantize_fp8_per_channel() : Mila::Dnn::Compute::Cuda::Linear::Detail
• quantize_fp8_per_tensor() : Mila::Dnn::Compute::Cuda::Linear::Detail

- r -

• readTensorBlob() : Mila::Dnn::Serialization
• registerBinaryOpType() : Mila::Dnn::Compute
• registerPairedOpType() : Mila::Dnn::Compute
• registerUnaryOpType() : Mila::Dnn::Compute
• renderHeatmapAvg() : Mila::Dnn::Visualization

- s -

• serializationModeToString() : Mila::Dnn::Serialization
• setCurrentDevice() : Mila::Dnn::Compute::Cuda
• shapeToString() : Mila::Dnn
• shutdown() : Mila
• split() : Mila::Dnn
• strideToString() : Mila::Dnn
• stringToActivationType() : Mila::Dnn
• stringToAttentionType() : Mila::Dnn
• stringToConnectionType() : Mila::Dnn
• stringToEncodingType() : Mila::Dnn
• stringToNormType() : Mila::Dnn
• stringToTokenizerType() : Mila::Data
• subtract() : Mila::Dnn
• sum() : Mila::Dnn

- t -

• tensorDataTypeToString() : Mila::Dnn
• toDeviceType() : Mila::Dnn::Compute
• toHost() : Mila::Dnn
• tokenizerTypeToString() : Mila::Data
• toString() : Mila::Data, Mila::Dnn
• toTypeId() : Mila::Dnn

- v -

• validateExecutionContext_() : Mila::Dnn::Compute
• validateTensorSize() : Mila::Dnn

- w -

• writeTensorBlob() : Mila::Dnn::Serialization

- z -

• zero() : Mila::Dnn