List of all items

Structs

• Matrix
• custos::Buffer
• custos::CPU
• custos::CUDA
• custos::Cache
• custos::CacheTrace
• custos::CachedLeaf
• custos::Count
• custos::CountIntoIter
• custos::Dim1
• custos::Dim2
• custos::Dim3
• custos::Graph
• custos::Ident
• custos::Node
• custos::Num
• custos::OpenCL
• custos::StackArray
• custos::devices::Cache
• custos::devices::Ident
• custos::devices::StackArray
• custos::devices::cache::Cache
• custos::devices::cpu::CPU
• custos::devices::cpu::CPUPtr
• custos::devices::cpu::RawCpuBuf
• custos::devices::cuda::CUDA
• custos::devices::cuda::CUDAPtr
• custos::devices::cuda::KernelCacheCU
• custos::devices::cuda::RawCUBuf
• custos::devices::cuda::api::Context
• custos::devices::cuda::api::CudaIntDevice
• custos::devices::cuda::api::FnHandle
• custos::devices::cuda::api::Module
• custos::devices::cuda::api::Stream
• custos::devices::cuda::api::cublas::CublasHandle
• custos::devices::cuda::api::nvrtc::NvrtcProgram
• custos::devices::opencl::CLDevice
• custos::devices::opencl::CLPtr
• custos::devices::opencl::KernelCacheCL
• custos::devices::opencl::OpenCL
• custos::devices::opencl::RawCL
• custos::devices::opencl::api::CLIntDevice
• custos::devices::opencl::api::CommandQueue
• custos::devices::opencl::api::Context
• custos::devices::opencl::api::DeviceReturnInfo
• custos::devices::opencl::api::Event
• custos::devices::opencl::api::Kernel
• custos::devices::opencl::api::Platform
• custos::devices::opencl::api::Program
• custos::devices::opencl::cl_device::OpenCL
• custos::prelude::Buffer
• custos::prelude::CPU
• custos::prelude::CUDA
• custos::prelude::Cache
• custos::prelude::Dim1
• custos::prelude::Dim2
• custos::prelude::Dim3
• custos::prelude::OpenCL

Enums

• custos::DeviceError
• custos::devices::cpu::Order
• custos::devices::cpu::Transpose
• custos::devices::cuda::api::CUctx_st
• custos::devices::cuda::api::CUdevice_attribute
• custos::devices::cuda::api::CUfunc_st
• custos::devices::cuda::api::CUmod_st
• custos::devices::cuda::api::CUresult
• custos::devices::cuda::api::CUstream_st
• custos::devices::cuda::api::cublas::cublasContext
• custos::devices::cuda::api::cublas::cublasOperation_t
• custos::devices::cuda::api::cublas::cublasStatus_t
• custos::devices::cuda::api::nvrtc::_nvrtcProgram
• custos::devices::cuda::api::nvrtc::nvrtcResult
• custos::devices::opencl::api::DeviceInfo
• custos::devices::opencl::api::DeviceType
• custos::devices::opencl::api::MemFlags
• custos::devices::opencl::api::OCLErrorKind
• custos::devices::opencl::api::PlatformInfo
• custos::flag::AllocFlag

Traits

• AdditionalOps
• AssignOps
• BaseOps
• ClipOp
• ColOp
• CudaTranspose
• DiagflatOp
• FnsOps
• Gemm
• MaxOps
• RandBuf
• RandOp
• RowOp
• ScalarAssign
• SumOps
• SumOverOps
• TransposeOp
• custos::AddGraph
• custos::Alloc
• custos::AsRangeArg
• custos::CDatatype
• custos::CacheAble
• custos::CacheBuf
• custos::CacheReturn
• custos::ClearBuf
• custos::CloneBuf
• custos::CommonPtrs
• custos::CopySlice
• custos::Device
• custos::DevicelessAble
• custos::ErrorKind
• custos::GenericBlas
• custos::GraphReturn
• custos::IsConstDim
• custos::IsShapeIndep
• custos::MainMemory
• custos::MayDim2
• custos::PtrType
• custos::RawConv
• custos::Read
• custos::ShallowCopy
• custos::Shape
• custos::ToDim
• custos::WithShape
• custos::WriteBuf
• custos::devices::CDatatype
• custos::devices::CacheAble
• custos::devices::CacheReturn
• custos::devices::GenericBlas
• custos::devices::RawConv
• custos::devices::cache::CacheReturn
• custos::devices::cache::RawConv
• custos::devices::cuda::AsCudaCvoidPtr
• custos::devices::opencl::AsClCvoidPtr
• custos::number::Float
• custos::number::Number
• custos::prelude::Alloc
• custos::prelude::CDatatype
• custos::prelude::CacheBuf
• custos::prelude::CacheReturn
• custos::prelude::ClearBuf
• custos::prelude::CopySlice
• custos::prelude::Device
• custos::prelude::Float
• custos::prelude::GraphReturn
• custos::prelude::IsConstDim
• custos::prelude::IsShapeIndep
• custos::prelude::MainMemory
• custos::prelude::MayDim2
• custos::prelude::Number
• custos::prelude::Read
• custos::prelude::ShallowCopy
• custos::prelude::Shape
• custos::prelude::ToDim
• custos::prelude::WithShape
• custos::prelude::WriteBuf
• custos::static_api::StaticGPU
• matrix_multiply::MatrixMultiply
• nn::ActivationOps
• nn::CCE
• nn::CCEOp
• nn::SoftmaxOps

Macros

• custos::buf

Attribute Macros

• custos::impl_stack

Functions

• cl_to_cpu_lr
• cl_to_cpu_s
• cpu::col_op
• cpu::col_op_slice_mut
• cpu::each_op
• cpu::each_op_slice
• cpu::each_op_slice_mut
• cpu::row_op
• cpu::row_op_slice_lhs
• cpu::row_op_slice_mut
• cpu::scalar_apply
• cpu::scalar_apply_slice
• cu_clip
• cuda::cu_scalar_op
• cuda::cu_str_op
• cuda::cu_str_op_mut
• cuda::cu_to_cpu_lr
• cuda::cu_to_cpu_lr_mut
• cuda::cu_to_cpu_s
• cuda::cu_to_cpu_s_mut
• cuda::cu_to_cpu_scalar
• custos::bump_count
• custos::cached
• custos::devices::bump_count
• custos::devices::cpu::api::cblas_dgemm
• custos::devices::cpu::api::cblas_sgemm
• custos::devices::cpu::cpu_cached
• custos::devices::cuda::api::cinit
• custos::devices::cuda::api::create_context
• custos::devices::cuda::api::create_stream
• custos::devices::cuda::api::cuCtxCreate_v2
• custos::devices::cuda::api::cuCtxDestroy
• custos::devices::cuda::api::cuCtxSynchronize
• custos::devices::cuda::api::cuDeviceGet
• custos::devices::cuda::api::cuDeviceGetAttribute
• custos::devices::cuda::api::cuDeviceGetCount
• custos::devices::cuda::api::cuInit
• custos::devices::cuda::api::cuLaunchKernel
• custos::devices::cuda::api::cuMemAlloc_v2
• custos::devices::cuda::api::cuMemFree_v2
• custos::devices::cuda::api::cuMemcpy
• custos::devices::cuda::api::cuMemcpyDtoH_v2
• custos::devices::cuda::api::cuMemcpyHtoD_v2
• custos::devices::cuda::api::cuModuleGetFunction
• custos::devices::cuda::api::cuModuleLoad
• custos::devices::cuda::api::cuModuleLoadData
• custos::devices::cuda::api::cuModuleUnload
• custos::devices::cuda::api::cuOccupancyMaxPotentialBlockSize
• custos::devices::cuda::api::cuStreamCreate
• custos::devices::cuda::api::cuStreamDestroy
• custos::devices::cuda::api::cuStreamSynchronize
• custos::devices::cuda::api::cu_read
• custos::devices::cuda::api::cu_write
• custos::devices::cuda::api::cublas::create_handle
• custos::devices::cuda::api::cublas::cublasCreate_v2
• custos::devices::cuda::api::cublas::cublasDestroy_v2
• custos::devices::cuda::api::cublas::cublasDgeam
• custos::devices::cuda::api::cublas::cublasDgemm_v2
• custos::devices::cuda::api::cublas::cublasSetStream_v2
• custos::devices::cuda::api::cublas::cublasSgeam
• custos::devices::cuda::api::cublas::cublasSgemm_v2
• custos::devices::cuda::api::cufree
• custos::devices::cuda::api::culaunch_kernel
• custos::devices::cuda::api::cumalloc
• custos::devices::cuda::api::device
• custos::devices::cuda::api::device_count
• custos::devices::cuda::api::load_module
• custos::devices::cuda::api::load_module_data
• custos::devices::cuda::api::module_get_fn
• custos::devices::cuda::api::nvrtc::compile_program
• custos::devices::cuda::api::nvrtc::create_program
• custos::devices::cuda::api::nvrtc::get_ptx
• custos::devices::cuda::api::nvrtc::nvrtcCompileProgram
• custos::devices::cuda::api::nvrtc::nvrtcCreateProgram
• custos::devices::cuda::api::nvrtc::nvrtcDestroyProgram
• custos::devices::cuda::api::nvrtc::nvrtcGetPTX
• custos::devices::cuda::api::nvrtc::nvrtcGetPTXSize
• custos::devices::cuda::chosen_cu_idx
• custos::devices::cuda::cu_cached
• custos::devices::cuda::cu_clear
• custos::devices::cuda::fn_cache
• custos::devices::cuda::launch_kernel1d
• custos::devices::get_count
• custos::devices::opencl::api::build_program
• custos::devices::opencl::api::clBuildProgram
• custos::devices::opencl::api::clCreateBuffer
• custos::devices::opencl::api::clCreateCommandQueue
• custos::devices::opencl::api::clCreateContext
• custos::devices::opencl::api::clCreateKernel
• custos::devices::opencl::api::clCreateKernelsInProgram
• custos::devices::opencl::api::clCreateProgramWithSource
• custos::devices::opencl::api::clEnqueueCopyBuffer
• custos::devices::opencl::api::clEnqueueFillBuffer
• custos::devices::opencl::api::clEnqueueMapBuffer
• custos::devices::opencl::api::clEnqueueNDRangeKernel
• custos::devices::opencl::api::clEnqueueReadBuffer
• custos::devices::opencl::api::clEnqueueWriteBuffer
• custos::devices::opencl::api::clFinish
• custos::devices::opencl::api::clGetDeviceIDs
• custos::devices::opencl::api::clGetDeviceInfo
• custos::devices::opencl::api::clGetPlatformIDs
• custos::devices::opencl::api::clGetPlatformInfo
• custos::devices::opencl::api::clGetProgramBuildInfo
• custos::devices::opencl::api::clGetProgramInfo
• custos::devices::opencl::api::clReleaseCommandQueue
• custos::devices::opencl::api::clReleaseContext
• custos::devices::opencl::api::clReleaseEvent
• custos::devices::opencl::api::clReleaseKernel
• custos::devices::opencl::api::clReleaseMemObject
• custos::devices::opencl::api::clReleaseProgram
• custos::devices::opencl::api::clRetainMemObject
• custos::devices::opencl::api::clSetKernelArg
• custos::devices::opencl::api::clWaitForEvents
• custos::devices::opencl::api::create_buffer
• custos::devices::opencl::api::create_command_queue
• custos::devices::opencl::api::create_context
• custos::devices::opencl::api::create_kernel
• custos::devices::opencl::api::create_kernels_in_program
• custos::devices::opencl::api::create_program_with_source
• custos::devices::opencl::api::enqueue_copy_buffer
• custos::devices::opencl::api::enqueue_full_copy_buffer
• custos::devices::opencl::api::enqueue_map_buffer
• custos::devices::opencl::api::enqueue_nd_range_kernel
• custos::devices::opencl::api::enqueue_read_buffer
• custos::devices::opencl::api::enqueue_write_buffer
• custos::devices::opencl::api::finish
• custos::devices::opencl::api::get_device_ids
• custos::devices::opencl::api::get_device_info
• custos::devices::opencl::api::get_platform_info
• custos::devices::opencl::api::get_platforms
• custos::devices::opencl::api::release_event
• custos::devices::opencl::api::release_kernel
• custos::devices::opencl::api::release_mem_object
• custos::devices::opencl::api::release_program
• custos::devices::opencl::api::retain_mem_object
• custos::devices::opencl::api::set_kernel_arg
• custos::devices::opencl::api::unified_ptr
• custos::devices::opencl::api::wait_for_event
• custos::devices::opencl::chosen_cl_idx
• custos::devices::opencl::cl_cached
• custos::devices::opencl::cl_clear
• custos::devices::opencl::cl_device::cl_cached
• custos::devices::opencl::enqueue_kernel
• custos::devices::set_count
• custos::get_count
• custos::prelude::cached
• custos::prelude::cpu_cached
• custos::prelude::enqueue_kernel
• custos::prelude::get_count
• custos::prelude::launch_kernel1d
• custos::prelude::range
• custos::prelude::set_count
• custos::range
• custos::set_count
• custos::static_api::static_cpu
• custos::static_api::static_cuda
• custos::static_api::static_opencl
• diagflat
• ew_op
• nn::cl_softmax
• nn::mse
• nn::mse_grad
• nn::mse_grad_cl
• nn::mse_loss
• opencl::cl_scalar_op_mat
• opencl::cl_str_op_mat
• opencl::cl_write
• opencl::cpu_exec
• opencl::cpu_exec_lhs_rhs
• opencl::cpu_exec_lhs_rhs_mut
• opencl::cpu_exec_mut
• opencl::cpu_exec_scalar
• rand_slice
• raw_ops::cpu::add_full_padding
• raw_ops::cpu::assign_to_lhs
• raw_ops::cpu::assign_to_lhs_scalar
• raw_ops::cpu::correlate_valid_mut
• raw_ops::cpu::element_wise_op
• raw_ops::cpu::element_wise_op_mut
• raw_ops::cpu::naive_gemm
• raw_ops::cpu::rot_kernel
• raw_ops::cuda::cu_assign_scalar
• raw_ops::cuda::cu_ew
• raw_ops::cuda::cu_ew_self
• raw_ops::opencl::cl_assign_scalar
• raw_ops::opencl::cl_diagflat
• raw_ops::opencl::cl_gemm
• raw_ops::opencl::cl_scalar_op
• raw_ops::opencl::cl_str_op
• raw_ops::opencl::cl_str_op_mut
• raw_ops::opencl::cl_tew
• raw_ops::opencl::cl_tew_self
• raw_ops::opencl::cl_transpose
• raw_prelude::assign_to_lhs
• raw_prelude::assign_to_lhs_scalar
• raw_prelude::cl_diagflat
• raw_prelude::cl_gemm
• raw_prelude::cl_scalar_op
• raw_prelude::cl_str_op
• raw_prelude::cl_tew
• raw_prelude::cl_tew_self
• raw_prelude::cl_transpose
• raw_prelude::rand_slice
• raw_prelude::slice_transpose
• slice_transpose

Type Definitions

• custos::CUdeviceptr
• custos::Error
• custos::Result
• custos::devices::CUdeviceptr
• custos::devices::cuda::CU
• custos::devices::cuda::CUBuffer
• custos::devices::cuda::api::CUcontext
• custos::devices::cuda::api::CUdevice
• custos::devices::cuda::api::CUfunction
• custos::devices::cuda::api::CUmodule
• custos::devices::cuda::api::CUstream
• custos::devices::cuda::api::cublas::cublasHandle_t
• custos::devices::cuda::api::nvrtc::nvrtcProgram
• custos::devices::opencl::CL
• custos::devices::opencl::CLBuffer
• custos::devices::opencl::Error
• custos::devices::opencl::api::cl_bitfield
• custos::devices::opencl::api::cl_bool
• custos::devices::opencl::api::cl_command_queue
• custos::devices::opencl::api::cl_command_queue_properties
• custos::devices::opencl::api::cl_context
• custos::devices::opencl::api::cl_context_properties
• custos::devices::opencl::api::cl_device_id
• custos::devices::opencl::api::cl_device_info
• custos::devices::opencl::api::cl_device_type
• custos::devices::opencl::api::cl_event
• custos::devices::opencl::api::cl_int
• custos::devices::opencl::api::cl_kernel
• custos::devices::opencl::api::cl_long
• custos::devices::opencl::api::cl_map_flags
• custos::devices::opencl::api::cl_mem
• custos::devices::opencl::api::cl_mem_flags
• custos::devices::opencl::api::cl_platform_id
• custos::devices::opencl::api::cl_platform_info
• custos::devices::opencl::api::cl_program
• custos::devices::opencl::api::cl_program_build_info
• custos::devices::opencl::api::cl_program_info
• custos::devices::opencl::api::cl_uint
• custos::devices::opencl::api::cl_ulong
• custos::devices::opencl::api::size_t
• custos::devices::opencl::cl_device::CL
• custos::prelude::CL
• custos::prelude::CLBuffer
• custos::prelude::CU
• custos::prelude::CUBuffer

Constants

• custos::COUNT
• custos::UNIFIED_CL_MEM
• custos::devices::COUNT
• custos::static_api::GLOBAL_CPU
• custos::static_api::GLOBAL_CUDA
• custos::static_api::GLOBAL_OPENCL