“42c1297e29c60ba3256648761392f8404209518b”上不存在“...tests/unittests/collective/parallel_embedding_api.py”
  • Q
    Refine device context (#6433) · 61ec0b95
    QI JUN 提交于
    There are mainly following fixes:
    
    - take `DeviceContext` as the template parameter of math functors and OpKernel instead of `Place`
    - remove `eigen_device` interface in base class  `DeviceContext`
    - remove `GetEigenDevice` interface in `ExecutionContext` and base class `DeviceContext`
    - remove unused `platform::EigenDeviceConverter`
    - rename `REGISTER_OP_GPU_KERNEL` to `REGISTER_OP_CUDA_KERNEL`
    - rename `USE_GPU_ONLY_OP` to `USE_CUDA_ONLY_OP`
    61ec0b95
math_function_test.cu 8.3 KB