infini::ops::GetCutlassMoeMmData Member List

This is the complete list of members for infini::ops::GetCutlassMoeMmData, including all inherited members.

active_implementation_indices(Device::Type dev_type)	infini::ops::Operator< GetCutlassMoeMmData >	inlinestatic
active_implementation_indices(Device::Type dev_type)	infini::ops::Operator< GetCutlassMoeMmData >	inlinestatic
blockscale_offsets_metadata_	infini::ops::GetCutlassMoeMmData	protected
Call(const Handle &handle, const Config &config, const Args &... args)	infini::ops::Operator< GetCutlassMoeMmData >	static
Call(const Tensor tensor, const Args &... args)	infini::ops::Operator< GetCutlassMoeMmData >	inlinestatic
Call(const TensorLike &tensor, const Args &... args)	infini::ops::Operator< GetCutlassMoeMmData >	inlinestatic
Call(const Handle &handle, const Config &config, const Args &... args)	infini::ops::Operator< GetCutlassMoeMmData >	inlinestatic
Call(const Tensor tensor, const Args &... args)	infini::ops::Operator< GetCutlassMoeMmData >	inlinestatic
Call(const TensorLike &tensor, const Args &... args)	infini::ops::Operator< GetCutlassMoeMmData >	inlinestatic
clear_cache()	infini::ops::Operator< GetCutlassMoeMmData >	inlinestatic
clear_cache()	infini::ops::Operator< GetCutlassMoeMmData >	inlinestatic
config_ptr_	infini::ops::OperatorBase	protected
device_index_	infini::ops::GetCutlassMoeMmData	protected
device_type_	infini::ops::Operator< GetCutlassMoeMmData >	protectedstatic
expert_offsets_metadata_	infini::ops::GetCutlassMoeMmData	protected
GetCutlassMoeMmData(const Tensor topk_ids, const int64_t num_experts, const int64_t n, const int64_t k, Tensor expert_offsets, Tensor problem_sizes1, Tensor problem_sizes2, Tensor input_permutation, Tensor output_permutation, std::optional< Tensor > blockscale_offsets=std::nullopt)	infini::ops::GetCutlassMoeMmData	inline
GetCutlassMoeMmData(const Tensor topk_ids, const int64_t num_experts, const int64_t n, const int64_t k, const bool is_gated, Tensor expert_offsets, Tensor problem_sizes1, Tensor problem_sizes2, Tensor input_permutation, Tensor output_permutation, std::optional< Tensor > blockscale_offsets=std::nullopt)	infini::ops::GetCutlassMoeMmData	inline
handle_ptr_	infini::ops::OperatorBase	protected
implementation_index_	infini::ops::Operator< GetCutlassMoeMmData >	protectedstatic
input_permutation_metadata_	infini::ops::GetCutlassMoeMmData	protected
is_gated_	infini::ops::GetCutlassMoeMmData	protected
k_	infini::ops::GetCutlassMoeMmData	protected
Make(const Config &config, const Tensor tensor, Args &&... args)	infini::ops::Operator< GetCutlassMoeMmData >	inlinestatic
Make(const Tensor tensor, Args &&... args)	infini::ops::Operator< GetCutlassMoeMmData >	inlinestatic
Make(const Config &config, const std::vector< Tensor > tensors, Args &&... args)	infini::ops::Operator< GetCutlassMoeMmData >	inlinestatic
Make(const std::vector< Tensor > tensors, Args &&... args)	infini::ops::Operator< GetCutlassMoeMmData >	inlinestatic
Make(const Config &config, const Tensor tensor, Args &&... args)	infini::ops::Operator< GetCutlassMoeMmData >	inlinestatic
Make(const Tensor tensor, Args &&... args)	infini::ops::Operator< GetCutlassMoeMmData >	inlinestatic
Make(const Config &config, const std::vector< Tensor > tensors, Args &&... args)	infini::ops::Operator< GetCutlassMoeMmData >	inlinestatic
Make(const std::vector< Tensor > tensors, Args &&... args)	infini::ops::Operator< GetCutlassMoeMmData >	inlinestatic
n_	infini::ops::GetCutlassMoeMmData	protected
num_experts_	infini::ops::GetCutlassMoeMmData	protected
numel_	infini::ops::GetCutlassMoeMmData	protected
operator()(const Tensor topk_ids, const int64_t num_experts, const int64_t n, const int64_t k, Tensor expert_offsets, Tensor problem_sizes1, Tensor problem_sizes2, Tensor input_permutation, Tensor output_permutation, std::optional< Tensor > blockscale_offsets=std::nullopt) const	infini::ops::GetCutlassMoeMmData	inline
operator()(const Tensor topk_ids, const int64_t num_experts, const int64_t n, const int64_t k, const bool is_gated, Tensor expert_offsets, Tensor problem_sizes1, Tensor problem_sizes2, Tensor input_permutation, Tensor output_permutation, std::optional< Tensor > blockscale_offsets=std::nullopt) const =0	infini::ops::GetCutlassMoeMmData	pure virtual
Operator< GetCutlassMoeMmData >::operator()(const Handle &handle, const Args &... args)	infini::ops::Operator< GetCutlassMoeMmData >	inline
Operator< GetCutlassMoeMmData >::operator()(const Args &... args) const	infini::ops::Operator< GetCutlassMoeMmData >	inline
output_permutation_metadata_	infini::ops::GetCutlassMoeMmData	protected
problem_sizes1_metadata_	infini::ops::GetCutlassMoeMmData	protected
problem_sizes2_metadata_	infini::ops::GetCutlassMoeMmData	protected
set_config(const Config &config)	infini::ops::OperatorBase	inline
set_config(const Config &config)	infini::ops::OperatorBase	inline
set_handle(const Handle &handle)	infini::ops::OperatorBase	inline
set_handle(const Handle &handle)	infini::ops::OperatorBase	inline
set_stream(void *stream)	infini::ops::OperatorBase	inline
set_stream(void *stream)	infini::ops::OperatorBase	inline
set_workspace(void *workspace)	infini::ops::OperatorBase	inline
set_workspace(void *workspace)	infini::ops::OperatorBase	inline
set_workspace_size_in_bytes(std::size_t workspace_size_in_bytes)	infini::ops::OperatorBase	inline
set_workspace_size_in_bytes(std::size_t workspace_size_in_bytes)	infini::ops::OperatorBase	inline
stream_	infini::ops::OperatorBase	protected
topk_	infini::ops::GetCutlassMoeMmData	protected
topk_ids_metadata_	infini::ops::GetCutlassMoeMmData	protected
ValidateCallMetadata(const Tensor topk_ids, const int64_t num_experts, const int64_t n, const int64_t k, const bool is_gated, const Tensor expert_offsets, const Tensor problem_sizes1, const Tensor problem_sizes2, const Tensor input_permutation, const Tensor output_permutation, const std::optional< Tensor > blockscale_offsets) const	infini::ops::GetCutlassMoeMmData	inlineprotected
workspace_	infini::ops::OperatorBase	protected
workspace_size_in_bytes() const	infini::ops::OperatorBase	inlinevirtual
workspace_size_in_bytes() const	infini::ops::OperatorBase	inlinevirtual
workspace_size_in_bytes_	infini::ops::OperatorBase	protected
~OperatorBase()=default	infini::ops::OperatorBase	virtual
~OperatorBase()=default	infini::ops::OperatorBase	virtual