MCPcopy Index your code
hub / github.com/hpcaitech/ColossalAI / assert_compatible

Method assert_compatible

extensions/base_extension.py:67–70  ·  view source on GitHub ↗

Check if the hardware required by the kernel is compatible.

(self)

Source from the content-addressed store, hash-verified

65
66 @abstractmethod
67 def assert_compatible(self) -> None:
68 """
69 Check if the hardware required by the kernel is compatible.
70 """
71
72 @abstractmethod
73 def build_aot(self) -> Union["CppExtension", "CUDAExtension"]:

Callers 3

setup.pyFile · 0.45
loadMethod · 0.45
test_flash_attn_funcFunction · 0.45

Calls

no outgoing calls

Tested by 1

test_flash_attn_funcFunction · 0.36