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()