hide some logs of p2p (#28307)

revert-28284-dev/pybind_version
Leo Chen 5 years ago committed by GitHub
parent 4086f48ea1
commit 18c86fb2fb
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

@ -12,6 +12,7 @@ WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
See the License for the specific language governing permissions and
limitations under the License. */
#include <string.h> // for strdup
#include <algorithm>
#include <fstream>
#include <iostream>
@ -105,8 +106,8 @@ void InitP2P(std::vector<int> devices) {
PADDLE_ENFORCE_CUDA_SUCCESS(
cudaDeviceCanAccessPeer(&can_acess, devices[i], devices[j]));
if (can_acess != 1) {
LOG(WARNING) << "Cannot enable P2P access from " << devices[i]
<< " to " << devices[j];
VLOG(2) << "Cannot enable P2P access from " << devices[i] << " to "
<< devices[j];
} else {
platform::CUDADeviceGuard guard(devices[i]);
cudaDeviceEnablePeerAccess(devices[j], 0);

@ -152,6 +152,9 @@ def __bootstrap__():
import platform
from . import core
# NOTE(zhiqiu): When (1)numpy < 1.19; (2) python < 3.7,
# unittest is always imported in numpy (maybe some versions not).
# so is_test is True and p2p is not inited.
in_test = 'unittest' in sys.modules
try:

Loading…
Cancel
Save