Function half

wan/distributed/xdit_context_parallel.py:193–194 · view source on GitHub ↗

(x)

Source from the content-addressed store, hash-verified

191	half_dtypes = (torch.float16, torch.bfloat16)
192
193	def half(x):
194	return x if x.dtype in half_dtypes else x.to(dtype)
195
196	# query, key, value function
197	def qkv_fn(x):

usp_attn_forwardFunction · 0.70

usp_attn_forward_multitalkFunction · 0.70

no outgoing calls

no test coverage detected