o i@s"ddlmZmZdefddZdS) CastConfigFloat8LinearConfigemulatecCs2t|d}t|d}t|d}t||||d}|S)N) scaling_type)cast_config_inputcast_config_weightcast_config_grad_outputrr)scaling_type_inputscaling_type_weightscaling_type_grad_outputrrrr configrW/home/ubuntu/.local/lib/python3.10/site-packages/torchao/testing/training/test_utils.pyget_test_float8_linear_config s rN)torchao.float8.configrrboolrrrrrs