Function half

wan/modules/attention.py:58–59 · view source on GitHub ↗

(x)

Source from the content-addressed store, hash-verified

56	b, lq, lk, out_dtype = q.size(0), q.size(1), k.size(1), q.dtype
57
58	def half(x):
59	return x if x.dtype in half_dtypes else x.to(dtype)
60
61	# preprocess query
62	if q_lens is None:

flash_attentionFunction · 0.70

toMethod · 0.80

no test coverage detected