tilelang.carver.roller.policy.common

函式

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