https://github.com/Microsoft/CNTK
Raw File
Tip revision: e9162e18e1c51b88f5c92ac6eb6623943eae353e authored by Amit Agarwal on 08 May 2016, 01:37:50 UTC
Synchronize batch norm nodes' running means and variances across parallel workers at end of each epoch to ensure accurate restartability across epochs and add and end-to-end test for this scenario
Tip revision: e9162e1
CPURNGHandle.cpp
//
// Copyright (c) Microsoft. All rights reserved.
// Licensed under the MIT license. See LICENSE.md file in the project root for full license information.
//
// CPUMatrix.cpp : full implementation of all matrix functions on the CPU side
//

#include "stdafx.h"
#include "CPURNGHandle.h"

namespace Microsoft { namespace MSR { namespace CNTK {

CPURNGHandle::CPURNGHandle(int deviceId, unsigned long seed)
    : RNGHandle(deviceId)
{
#ifdef _MSC_VER // TODO: check if available under GCC/Linux
    m_generator.reset(new std::ranlux64_base_01());
    m_generator->seed(seed);
#else
    m_generator.reset(new std::default_random_engine(seed));
#endif
}

}}}
back to top