CUTLASS Python API

• CUTLASS
  • Subpackages
    • Emitters
      • Common
      • PyTorch
        • pytorch()
    • Operations
      • GEMM
        • Gemm
      • Grouped GEMM
        • GroupedGemm
      • Operation
        • OperationBase
    • Utilities
      • Checks
        • alignment_or_default()
        • calculate_smem_usage()
        • calculate_smem_usage_per_stage()
        • valid_cluster_shape()
        • valid_kernel_schedule()
        • valid_stage_count()
      • Data Types
        • backend_math_operation()
        • bfloat16_library_type()
        • bfloat16_type()
        • binding_layout()
        • binding_library_type()
        • binding_opclass()
        • binding_type()
        • construct_backend_td()
        • cupy_library_type()
        • cupy_type()
        • get_datatype_and_layout()
        • has_binding_type()
        • library_layout()
        • library_to_binding()
        • library_type()
        • numpy_library_type()
        • numpy_type()
        • td_from_profiler_op()
        • td_from_profiler_td()
        • torch_library_type()
        • torch_type()
  • Epilogue
    • get_activation_epilogue()
    • get_activations()
  • Library Defaults
    • ArchOptions
      • ArchOptions.opclass_supports_combination()
      • ArchOptions.operations()
      • ArchOptions.supporting_opclasses()
    • KernelsForDataType
      • KernelsForDataType.add()
      • KernelsForDataType.alignments
      • KernelsForDataType.all_operations
      • KernelsForDataType.find_alignment()
      • KernelsForDataType.operations()
      • KernelsForDataType.sort()
    • OptionRegistry
      • OptionRegistry.options_for_cc()
  • Swizzle
    • get_swizzling_functors()