Skip to main content

Browse the archive

https://github.com/Microsoft/CNTK
10 March 2023, 01:01:58 UTC
Revision 7877e3ffa0785839ac9462643829a410936de71c authored by Chengji Yao on 31 January 2018, 03:18:00 UTC, committed by Chengji Yao on 01 February 2018, 02:22:27 UTC 
add hierarchical softmax
1 parent abfc3a8
Raw File
Tip revision: 7877e3ffa0785839ac9462643829a410936de71c authored by Chengji Yao on 31 January 2018, 03:18:00 UTC
add hierarchical softmax
Tip revision: 7877e3f
GPUWatcher.h 
//
// Copyright (c) Microsoft. All rights reserved.
// Licensed under the MIT license. See LICENSE.md file in the project root for full license information.
//

#pragma once

#include "GPUMatrix.h"

class MATH_API GPUWatcher
{
public:
    static size_t GetFreeMemoryOnCUDADevice(int devId);
    static int GetGPUIdWithTheMostFreeMemory();
    GPUWatcher(void);
    ~GPUWatcher(void);
};
back to top