experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Remove warning that only happening with config cuda.

Browse Source 
Change: 152189205
This commit is contained in:
Suharsh Sivakumar 2017-04-04 14:36:38 -08:00 committed by TensorFlower Gardener
parent 64ca47409c
commit a42b3fc598
2 changed files with 6 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
tensorflow/core/common_runtime/gpu/gpu_device.cc
View File

@ -274,14 +274,14 @@ Status BaseGPUDevice::FillContextMap(const Graph* graph,
DeviceContextMap* device_context_map) { DeviceContextMap* device_context_map) {
VLOG(2) << "FillContextMap"; VLOG(2) << "FillContextMap";
const auto num_streams = streams_.size(); const size_t num_streams = streams_.size();
// Special case for single stream. // Special case for single stream.
if (num_streams == 1) { if (num_streams == 1) {
return Status::OK(); return Status::OK();
} }
const int64 before = Env::Default()->NowMicros(); const int64 before = Env::Default()->NowMicros();
gpu_stream_util::AssignStreamsOpts opts; gpu_stream_util::AssignStreamsOpts opts;
opts.max_streams = num_streams; opts.max_streams = static_cast<int32>(num_streams);
std::unordered_map<int, int> node_to_stream_id; std::unordered_map<int, int> node_to_stream_id;
TF_RETURN_IF_ERROR( TF_RETURN_IF_ERROR(
gpu_stream_util::AssignStreams(graph, opts, &node_to_stream_id)); gpu_stream_util::AssignStreams(graph, opts, &node_to_stream_id));
@ -519,7 +519,7 @@ void BaseGPUDevice::ReinitializeGpuDevice(OpKernelContext* context,
Status BaseGPUDeviceFactory::CreateDevices(const SessionOptions& options, Status BaseGPUDeviceFactory::CreateDevices(const SessionOptions& options,
const string& name_prefix, const string& name_prefix,
std::vector<Device*>* devices) { std::vector<Device*>* devices) {
int n = INT_MAX; size_t n = INT_MAX;
auto iter = options.config.device_count().find("GPU"); auto iter = options.config.device_count().find("GPU");
if (iter != options.config.device_count().end()) { if (iter != options.config.device_count().end()) {
n = iter->second; n = iter->second;
@ -971,7 +971,7 @@ Status BaseGPUDeviceFactory::GetValidDeviceIds(
continue; continue;
} }
int new_id = ids->size(); size_t new_id = ids->size();
ids->push_back(visible_gpu_id); ids->push_back(visible_gpu_id);
LOG(INFO) << "Creating TensorFlow device (/gpu:" << new_id << ") -> " LOG(INFO) << "Creating TensorFlow device (/gpu:" << new_id << ") -> "

4
tensorflow/core/common_runtime/gpu/gpu_event_mgr_test.cc
View File

@ -37,12 +37,12 @@ class TEST_EventMgrHelper {
StopPollingLoop(); StopPollingLoop();
} }
int queue_size() { size_t queue_size() {
mutex_lock l(em_->mu_); mutex_lock l(em_->mu_);
return em_->used_events_.size(); return em_->used_events_.size();
} }
int free_size() { size_t free_size() {
mutex_lock l(em_->mu_); mutex_lock l(em_->mu_);
return em_->free_events_.size(); return em_->free_events_.size();
} }