Mila 0.13.48
Deep Neural Network Library
Loading...
Searching...
No Matches
CudaSwigluOp.Dispatch.ixx File Reference
#include <vector>
#include <memory>
#include <iostream>
#include <cuda_bf16.h>
#include <stdexcept>
#include <type_traits>
#include <string>
#include "Kernels/Swiglu.cuh"
import Compute.CudaDevice;
import Compute.CudaTensorDataType;
import Compute.MemoryResource;
import Dnn.Components.SwigluConfig;
import Dnn.Tensor;
import Compute.OperationRegistry;
import Dnn.TensorDataType;
import Dnn.TensorDataTypeTraits;
import Compute.ExecutionContextTemplate;
import Dnn.ComponentConfig;
import Compute.OperationType;
import Dnn.ITensor;
import Compute.DeviceType;
import Compute.CudaDeviceMemoryResource;
import Compute.OperationBase;
import Compute.IExecutionContext;
import Compute.UnaryOperation;
import Compute.ExecutionContext;

Classes

struct  Mila::Dnn::Compute::Cuda::Swiglu::Detail::cuda_swiglu_impl< __nv_bfloat16 >
struct  Mila::Dnn::Compute::Cuda::Swiglu::Detail::cuda_swiglu_impl< float >

Namespaces

namespace  Mila
 Mila main API namespace.
namespace  Mila::Dnn
namespace  Mila::Dnn::Compute
namespace  Mila::Dnn::Compute::Cuda
namespace  Mila::Dnn::Compute::Cuda::Swiglu
namespace  Mila::Dnn::Compute::Cuda::Swiglu::Detail