o }oi@sddlmZGdddZdS)) lightningc@seZdZddZdS)TestFabricConversioncCsrtjddtjddtjddd}|}t|jtjsJ|jjdks&Jt|j tj s/J|j j dks7JdS)Ncpu)tensor_model_parallel_sizez16-mixed) precision)devices acceleratorstrategyplugins) nlTrainerMegatronStrategyMegatronMixedPrecision to_fabric isinstancer FabricMegatronStrategyr _precisionFabricMegatronMixedPrecisionr)selftrainerfabricrX/home/ubuntu/.local/lib/python3.10/site-packages/tests/lightning/pytorch/test_trainer.pytest_simple_conversions  z+TestFabricConversion.test_simple_conversionN)__name__ __module__ __qualname__rrrrrrs rN)nemorr rrrrrs