tilelang.carver.roller.policy.common

Functions

get_all_factors(n)
factorize(n)
coalesced_factor(subtensor, tensor)
coalesced_tensor_shape(subtensor, tensor, transaction_size)

Module Contents

tilelang.carver.roller.policy.common.get_all_factors(n)

参数:

n (int)

返回类型:

list[int]

tilelang.carver.roller.policy.common.factorize(n)

参数:

n (int)

返回类型:

list[int]

tilelang.carver.roller.policy.common.coalesced_factor(subtensor, tensor)

参数:

• subtensor (list[int])

• tensor (list[int])

返回类型:

int

tilelang.carver.roller.policy.common.coalesced_tensor_shape(subtensor, tensor, transaction_size)

参数:

• subtensor (list[int])

• tensor (list[int])

• transaction_size (int)

返回类型:

int