Team Ai
Apppublic

CVPR/LIVE

sourceHugging Facegpl-3.0updated 4y agoView on Hugging Face
34likes
atomic.cpp27 linesDownload Raw Back to root
1//A hacky solution to get around the Ellipse include2 3#ifdef WIN324#include <windows.h>5#include <cstdint>6 7float win_atomic_add(float &target, float source) {8	union { int i; float f; } old_val;9	union { int i; float f; } new_val;10	do {11		old_val.f = target;12		new_val.f = old_val.f + (float)source;13	} while (InterlockedCompareExchange((LONG*)&target, (LONG)new_val.i, (LONG)old_val.i) != old_val.i);14	return old_val.f;15}16 17double win_atomic_add(double &target, double source) {18	union { int64_t i; double f; } old_val;19	union { int64_t i; double f; } new_val;20	do {21		old_val.f = target;22		new_val.f = old_val.f + (double)source;23	} while (InterlockedCompareExchange64((LONG64*)&target, (LONG64)new_val.i, (LONG64)old_val.i) != old_val.i);24	return old_val.f;25}26 27#endif