tilelang.testing

Functions

main()

set_random_seed([seed])

requires_cuda_compute_version(major_version[, ...])

Mark a test as requiring at least a compute architecture

requires_cuda_compute_version_ge(major_version[, ...])

requires_cuda_compute_version_gt(major_version[, ...])

requires_cuda_compute_version_eq(major_version[, ...])

requires_cuda_compute_version_lt(major_version[, ...])

requires_cuda_compute_version_le(major_version[, ...])

Package Contents

tilelang.testing.main()
tilelang.testing.set_random_seed(seed=42)
Parameters:

seed (int)

Return type:

None

tilelang.testing.requires_cuda_compute_version(major_version, minor_version=0, mode='ge')

Mark a test as requiring at least a compute architecture

Unit test marked with this decorator will run only if the CUDA compute architecture of the GPU is at least (major_version, minor_version).

This also marks the test as requiring a cuda support.

Parameters:

  • major_version (int) – The major version of the (major,minor) version tuple.

  • minor_version (int) – The minor version of the (major,minor) version tuple.

  • mode (str) – The mode of the comparison. - “ge”: greater than or equal to - “gt”: greater than - “le”: less than or equal to - “lt”: less than

tilelang.testing.requires_cuda_compute_version_ge(major_version, minor_version=0)
tilelang.testing.requires_cuda_compute_version_gt(major_version, minor_version=0)
tilelang.testing.requires_cuda_compute_version_eq(major_version, minor_version=0)
tilelang.testing.requires_cuda_compute_version_lt(major_version, minor_version=0)
tilelang.testing.requires_cuda_compute_version_le(major_version, minor_version=0)