We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent ae94669 commit 02e90d5Copy full SHA for 02e90d5
src/cuda/causal_softmax/kernel.h
@@ -36,7 +36,7 @@ class CudaCausalSoftmax : public CausalSoftmax {
36
37
DispatchFunc<ConcatType<List<DataType::kFloat32>, ReducedFloatTypes>,
38
AllCudaBlockSizes>(
39
- // TODO: output dtype should use the one passed in during construction
+ // TODO: output dtype should use the one passed in during construction.
40
{static_cast<int64_t>(out.dtype()), block_size},
41
[&](auto list_tag) {
42
using T = TypeMapType<ListGet<0>(list_tag)>;
0 commit comments