#define THREADS 64 #include "softMax.hlsl"