Function next_power_of_2

src/fused_mm_sampling/qitra.py:23–24 · view source on GitHub ↗

(n: int)

Source from the content-addressed store, hash-verified

21
22
23	def next_power_of_2(n: int) -> int:
24	return 1 << math.ceil(math.log2(max(n, 1)))
25
26
27	def num_compute_units(device_index: int \| None = None) -> int:

apply_top_k_top_p_tritonFunction · 0.85

no outgoing calls

no test coverage detected