MCPcopy Create free account
hub / github.com/AnswerDotAI/gpu.cpp / SoftmaxParam

Class SoftmaxParam

experimental/kernels/ops.cpp:376–380  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

374
375void softmax_forward(Context& ctx, float* probs, float* logits, int B, int T, int V, int Vp) {
376 struct SoftmaxParam {
377 uint32_t N;
378 uint32_t C;
379 uint32_t Cp;
380 };
381 uint32_t b = static_cast<uint32_t>(B);
382 uint32_t t = static_cast<uint32_t>(T);
383 uint32_t c = static_cast<uint32_t>(V);

Callers

nothing calls this directly

Calls

no outgoing calls

Tested by

no test coverage detected