Skip to content

Commit 693a36f

Browse files
fix docstring
1 parent 0a36651 commit 693a36f

File tree

2 files changed

+2
-1
lines changed

2 files changed

+2
-1
lines changed

src/layers/recurrent.jl

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -237,6 +237,7 @@ julia> h′, c′ = l(x, (h, c));
237237
238238
julia> size(h′) # out x batch_size
239239
(5, 4)
240+
```
240241
"""
241242
struct LSTMCell{I,H,V}
242243
Wi::I

test/runtests.jl

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -11,7 +11,7 @@ using Functors: fmapstructure_with_path
1111

1212
## Uncomment below to change the default test settings
1313
# ENV["FLUX_TEST_AMDGPU"] = "true"
14-
# ENV["FLUX_TEST_CUDA"] = "true"
14+
ENV["FLUX_TEST_CUDA"] = "true"
1515
# ENV["FLUX_TEST_METAL"] = "true"
1616
# ENV["FLUX_TEST_CPU"] = "false"
1717
# ENV["FLUX_TEST_DISTRIBUTED_MPI"] = "true"

0 commit comments

Comments
 (0)