)`iddlmZddlZddlmZmZdejfdZdejfdZdejdejfd Z dd eejd ee d ee fd Z dS))ListN)get_compute_capabilityGPUArchitectureErrorxc|jsJt|d|sJt|ddS)Nz must be a CUDA Tensorz must be contiguous)is_cudastr is_contiguous)rs k/home/jaya/work/projects/VOICE-AGENT/VIET/agent-env/lib/python3.11/site-packages/flashinfer/triton/utils.py check_inputr sY 977Q777779 ??  <<Q<<<<< <<cp||ksJt|d|ddS)Nz must be a zD tensor)dimr )drs r check_dimr s8 5577a<<r)s IIIIIIII=5<==== ;EL;;;; 5< EL    NR %, (,S AEcr