"examples/transformer-xl/train.py" did not exist on "37d01e9c50d95c3f6be143a1242a6076b0bae3b5"
gpu_info_darwin.m 234 Bytes
Newer Older
1
2
3
4
5
6
7
8
9
10
11
//go:build darwin
#include "gpu_info_darwin.h"

uint64_t getRecommendedMaxVRAM()
{
	id<MTLDevice> device = MTLCreateSystemDefaultDevice();
	uint64_t result = device.recommendedMaxWorkingSetSize;
	CFRelease(device);
	return result;
}