https://github.com/Microsoft/CNTK
Raw File
Tip revision: ab4bee2b7a16a3e47379db140d5c55e31882ad7b authored by liqfu on 08 November 2018, 02:36:20 UTC
Support RNN ops in a Scan loop
Tip revision: ab4bee2
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