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)

Parameters:

n (int)

Return type:

List[int]

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

Parameters:

n (int)

Return type:

List[int]

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

Parameters:

• subtensor (List[int])

• tensor (List[int])

Return type:

int

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

Parameters:

• subtensor (List[int])

• tensor (List[int])

• transaction_size (int)

Return type:

int