tilelang.cuda.codegen

Attributes

build_cuda
build_cuda_without_compile
build_cutedsl
build_cutedsl_without_compile

Functions

is_cutedsl_target(target)
is_plain_cuda_target(target)

Module Contents

tilelang.cuda.codegen.is_cutedsl_target(target)

Parameters:

target (tvm.target.Target)

Return type:

bool

tilelang.cuda.codegen.is_plain_cuda_target(target)

Parameters:

target (tvm.target.Target)

Return type:

bool

tilelang.cuda.codegen.build_cuda

tilelang.cuda.codegen.build_cuda_without_compile

tilelang.cuda.codegen.build_cutedsl

tilelang.cuda.codegen.build_cutedsl_without_compile