• TaoTao Li's avatar
    Add comm context manager, add phi broadcast op (#51072) · c191b707
    TaoTao Li 提交于
    * * add comm context for device context
    
    * add broadcast phi operator kernel and api
    
    * add broadcast support dtype, update ut
    
    * fix broadcast bfloat16 type
    
    * fix ut
    
    * update test_collective_broadcast_api timeout to 300
    c191b707
device_context.h 5.4 KB