Function supports_tma

src/fused_mm_sampling/persistent_matmul.py:39–40 · view source on GitHub ↗

()

Source from the content-addressed store, hash-verified

37
38
39	def supports_tma():
40	return is_cuda() and torch.cuda.get_device_capability()[0] >= 9
41
42
43	def is_hopper():

persistent_matmul.pyFile · 0.85

is_cudaFunction · 0.85

no test coverage detected