megatron.initialize.set_jit_fusion_options#

megatron.initialize.set_jit_fusion_options(args)#

Set PyTorch JIT layer fusion options.