/jFddlZddlmZmZdedejjddfdZdS)N) OpOverloadOpOverloadPacketopgraphreturnct|trtd|t|tstdt |t j|j|dS)Nz;Must pass specific op overload, not overload packet, found zExpected OpOverload, got ) isinstancerAssertionErrorrtypetorch_C&_jit_register_decomposition_for_schema_schema)rrs c/home/longshao/multi-rider-rag/.venv/lib/python3.11/site-packages/torch/jit/_decomposition_utils.py_register_decompositionrs"&''  N" N N    b* % %ECbCCDDD H33BJFFFFF)r torch._opsrrr Graphrrrrsf 33333333G G58>GdGGGGGGr