o پi@s`ddlmZddlmZddlZddlmZmZmZer"ddl m Z edd d Z dddZ dS)) annotations) TYPE_CHECKINGN) cache_onceload_jit make_cpp_args)ModuleconstantintreturnrcCs0t|}tdg|Rdgdd|dfgdS)N add_constantzadd_constant.cuhz add_constant<>) cuda_files cuda_wrappers)rr)rargsrR/home/ubuntu/.local/lib/python3.10/site-packages/sglang/jit_kernel/add_constant.py_jit_add_constant_module srsrc torch.TensorcCs"t|}t|}||||S)N)torch empty_likerr )rrdstmodulerrrr s  r )rr r r)rrrr r r) __future__rtypingrrsglang.jit_kernel.utilsrrrtvm_ffi.modulerrr rrrrs