Searched refs:testAllreduceUsingWorkAPI (Results 1 – 1 of 1) sorted by relevance
307 void testAllreduceUsingWorkAPI( in testAllreduceUsingWorkAPI() function750 testAllreduceUsingWorkAPI(file.path, at::DeviceType::CPU); in TEST()758 testAllreduceUsingWorkAPI(file.path, at::DeviceType::CPU); in TEST()836 testAllreduceUsingWorkAPI(file.path, at::DeviceType::CUDA); in TEST()