https://github.com/Microsoft/CNTK
Raw File
Tip revision: 5666c0d470fdf8c708a9802e6074117795ea0596 authored by KeDengMS on 31 August 2018, 20:42:29 UTC
Upgrade to CUDA 9.2, CUDNN 7.2, NCCL 2.2.13, CUB 1.8.0
Tip revision: 5666c0d
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