o
    i)                     @   sd   U d dl Z i Zee jjee f ed< eddD ]Z	de	 dZ
de
 dgeee jje
j< qdS )	    Ntorchao_op_c_shim      _linear_fp_act_
bit_weightzAOTITorchError aoti_torch_mps_z{(AtenTensorHandle A, AtenTensorHandle B, int64_t group_size, AtenTensorHandle S, AtenTensorHandle Z, AtenTensorHandle* ret))torchr   dictops
OpOverloadliststr__annotations__rangenbitop_namegetattrtorchaodefault r   r   V/home/ubuntu/.local/lib/python3.10/site-packages/torchao/experimental/ops/mps/cshim.py<module>   s   
 
