Index

A | B | C | F | G | H | K | L | M | N | O | P | R | S | T | V

A

|

• activation (cutlass.op.gemm.Gemm property)
• activations() (cutlass.op.op.OperationBase method)
• add() (cutlass.library_defaults.KernelsForDataType method)

|

• alignment_or_default() (in module cutlass.utils.check)
• alignments (cutlass.library_defaults.KernelsForDataType property)
• all_operations (cutlass.library_defaults.KernelsForDataType property)
• ArchOptions (class in cutlass.library_defaults)

|

B

|

• backend_math_operation() (in module cutlass.utils.datatypes)
• bfloat16_library_type() (in module cutlass.utils.datatypes)
• bfloat16_type() (in module cutlass.utils.datatypes)

|

• binding_layout() (in module cutlass.utils.datatypes)
• binding_library_type() (in module cutlass.utils.datatypes)
• binding_opclass() (in module cutlass.utils.datatypes)
• binding_type() (in module cutlass.utils.datatypes)

|

C

|

• calculate_smem_usage() (in module cutlass.utils.check)

• calculate_smem_usage_per_stage() (in module cutlass.utils.check)

• compile() (cutlass.op.gemm.Gemm method)

• construct() (cutlass.op.gemm.Gemm method)

  • (cutlass.op.gemm_grouped.GroupedGemm method)

• construct_backend_td() (in module cutlass.utils.datatypes)

• cupy_library_type() (in module cutlass.utils.datatypes)

• cupy_type() (in module cutlass.utils.datatypes)

• cutlass.emit.common

  • module

• cutlass.emit.pytorch

  • module

• cutlass.epilogue

  • module

|

• cutlass.library_defaults

  • module

• cutlass.op.gemm

  • module

• cutlass.op.gemm_grouped

  • module

• cutlass.op.op

  • module

• cutlass.swizzle

  • module

• cutlass.utils.check

  • module

• cutlass.utils.datatypes

  • module

|

F

|

• find_alignment() (cutlass.library_defaults.KernelsForDataType method)

|

G

|

• Gemm (class in cutlass.op.gemm)
• get_activation_epilogue() (in module cutlass.epilogue)
• get_activations() (in module cutlass.epilogue)

|

• get_datatype_and_layout() (in module cutlass.utils.datatypes)
• get_swizzling_functors() (in module cutlass.swizzle)
• GroupedGemm (class in cutlass.op.gemm_grouped)

|

H

|

• has_binding_type() (in module cutlass.utils.datatypes)

|

K

|

• KernelsForDataType (class in cutlass.library_defaults)

|

L

|

• library_layout() (in module cutlass.utils.datatypes)

|

• library_to_binding() (in module cutlass.utils.datatypes)
• library_type() (in module cutlass.utils.datatypes)

|

M

|

• module
  • cutlass.emit.common
  • cutlass.emit.pytorch
  • cutlass.epilogue
  • cutlass.library_defaults
  • cutlass.op.gemm
  • cutlass.op.gemm_grouped
  • cutlass.op.op
  • cutlass.swizzle
  • cutlass.utils.check
  • cutlass.utils.datatypes

|

N

|

• numpy_library_type() (in module cutlass.utils.datatypes)

|

• numpy_type() (in module cutlass.utils.datatypes)

|

O

|

• opclass (cutlass.op.gemm.Gemm property)
• opclass_supports_combination() (cutlass.library_defaults.ArchOptions method)
• OperationBase (class in cutlass.op.op)

|

• operations() (cutlass.library_defaults.ArchOptions method)

  • (cutlass.library_defaults.KernelsForDataType method)

• OptionRegistry (class in cutlass.library_defaults)

• options_for_cc() (cutlass.library_defaults.OptionRegistry method)

|

P

|

• pytorch() (in module cutlass.emit.pytorch)

|

R

|

• run() (cutlass.op.gemm.Gemm method)
  • (cutlass.op.gemm_grouped.GroupedGemm method)

|

S

|

• sort() (cutlass.library_defaults.KernelsForDataType method)
• supporting_opclasses() (cutlass.library_defaults.ArchOptions method)

|

• swizzling_functor (cutlass.op.gemm.Gemm property)

  • (cutlass.op.gemm_grouped.GroupedGemm property)

• swizzling_functors() (cutlass.op.op.OperationBase method)

|

T

|

• td_from_profiler_op() (in module cutlass.utils.datatypes)
• td_from_profiler_td() (in module cutlass.utils.datatypes)

|

• tile_descriptions() (cutlass.op.gemm.Gemm method)
• torch_library_type() (in module cutlass.utils.datatypes)
• torch_type() (in module cutlass.utils.datatypes)

|

V

|

• valid_cluster_shape() (in module cutlass.utils.check)

|

• valid_kernel_schedule() (in module cutlass.utils.check)
• valid_stage_count() (in module cutlass.utils.check)

|