
    Mi                         S SK r S SKJr  \R                  S:X  a
  S SKJs  Jr  OSr\R                  4S\	S\	4S jjr
S\S\4S	 jrS
 rg)    N)settingsxlalengthpad_multiplec                 8    [         R                  " X-  5      U-  $ N)mathceil)r   r   s     j/var/www/eduai.edurigo.com/storigo/production/storigo_env/lib/python3.13/site-packages/surya/common/xla.pyget_nearest_padr   
   s     99V*+l::    devicereturnc                 8    [         R                  (       d  0 $ SS0$ )Nbackendopenxla)r   FOUNDATION_XLA)r   s    r   get_compile_argsr      s!    ""	 	9 r   c                  >    [         b  [         R                  " 5         g g r   )xm	mark_step r   r   r   r      s    	~
 r   )r	   surya.settingsr   TORCH_DEVICE_MODELtorch_xla.core.xla_modelcore	xla_modelr   FOUNDATION_PAD_TO_NEARESTintr   strdictr   r   r   r   r   <module>r"      s\     #%'))	B &.%G%G;;";S T r   