o ܹiT@snddlmZddlmZdgZddZde_eddZeddd Zed d Z ed d Z eddZ dS))driver)constexpr_functioncurrent_targetcCs&ztj}W|StyYdSw)N)ractive RuntimeErrorget_current_target) active_driverr X/home/ubuntu/veenaModal/venv/lib/python3.10/site-packages/triton/language/target_info.pyrs  TcCt}|duo |jdkS)Ncudarbackendtargetr r r is_cudarcCs>t}|dus |jdkrdSt|jtsJ|j|d|kS)z Determines whether we have compute capability >= (major, minor) and returns this as a constexpr boolean. This can be used for guarding inline asm implementations that require a certain compute capability. Nr F )rr isinstancearchint)majorminorrr r r cuda_capability_geqs rcCr )Nhipr rr r r is_hip'rrcCr )Ngfx942rrrr r r is_hip_cdna3-rrcCr )Ngfx950rrr r r is_hip_cdna43rr N)r) triton.runtimertriton.runtime.jitr__all__r__triton_builtin__rrrrr r r r r s