https://github.com/Microsoft/CNTK
Raw File
Tip revision: c18212eeac755c586f2dbf29fac00eee47bba1f4 authored by Peyman Manikashani on 02 August 2018, 23:13:08 UTC
BN FP16 fix
Tip revision: c18212e
CPUMatrixFloat.cpp
//
// Copyright (c) Microsoft. All rights reserved.
// Licensed under the MIT license. See LICENSE.md file in the project root for full license information.
//
#include "stdafx.h"
#include "CPUMatrixImpl.h"

namespace Microsoft { namespace MSR { namespace CNTK {

    int MATH_API TracingGPUMemoryAllocator::m_traceLevel = 0;

    void TracingGPUMemoryAllocator::SetTraceLevel(int traceLevel)
    {
        m_traceLevel = traceLevel;
    }

    bool TracingGPUMemoryAllocator::IsTraceEnabled()
    {
        return (m_traceLevel > 0);
    }

    // explicit instantiations, due to CPUMatrix being too big and causing VS2015 cl crash.
    template class MATH_API CPUMatrix<float>;
    template<> int CPUMatrix<float>::m_optimizationFlags = CPUMatrix<float>::OPT_EVAL_WITH_MKL; // enable eval MKL optimization by default
}}}
back to top