From 2bb25988006d7e41cbd38fa7a3f35881b825d4a9 Mon Sep 17 00:00:00 2001 From: chengduoZH Date: Tue, 12 Dec 2017 14:03:11 +0800 Subject: [PATCH] fix nvprof gpu consistence --- paddle/resnet.py | 1 + paddle/understand_sentiment_dynamic_lstm.py | 1 + paddle/understand_sentiment_lstm.py | 1 + 3 files changed, 3 insertions(+) diff --git a/paddle/resnet.py b/paddle/resnet.py index 620b0bc..6c62dc7 100644 --- a/paddle/resnet.py +++ b/paddle/resnet.py @@ -63,6 +63,7 @@ def parse_args(): def print_arguments(args): + vars(args)['use_nvprof'] = (vars(args)['use_nvprof'] and vars(args)['device']=='GPU') print('----------- Configuration Arguments -----------') for arg, value in sorted(vars(args).iteritems()): print('%s: %s' % (arg, value)) diff --git a/paddle/understand_sentiment_dynamic_lstm.py b/paddle/understand_sentiment_dynamic_lstm.py index 93e9337..80caa51 100644 --- a/paddle/understand_sentiment_dynamic_lstm.py +++ b/paddle/understand_sentiment_dynamic_lstm.py @@ -47,6 +47,7 @@ def parse_args(): def print_arguments(args): + vars(args)['use_nvprof'] = (vars(args)['use_nvprof'] and vars(args)['device']=='GPU') print('----------- Configuration Arguments -----------') for arg, value in sorted(vars(args).iteritems()): print('%s: %s' % (arg, value)) diff --git a/paddle/understand_sentiment_lstm.py b/paddle/understand_sentiment_lstm.py index c9d4dd1..7bb5bf0 100644 --- a/paddle/understand_sentiment_lstm.py +++ b/paddle/understand_sentiment_lstm.py @@ -47,6 +47,7 @@ def parse_args(): def print_arguments(args): + vars(args)['use_nvprof'] = (vars(args)['use_nvprof'] and vars(args)['device']=='GPU') print('----------- Configuration Arguments -----------') for arg, value in sorted(vars(args).iteritems()): print('%s: %s' % (arg, value))