JuliaGPU / JuliaGPU/Metal.jl

MPS: `randn!` produces NaNs

Open
#474 14 comments 0 reactions 0 assignees View on GitHub
upstream
Dominant language
Julia
Stars
463
Forks
68
Avg merge
1d 19m
Merged PRs (30d)
32

Description

Hi,

On a M2 Max with

```
(code) pkg> st
[e9467ef8] GLMakie v0.10.15
[63c18a36] KernelAbstractions v0.9.29
[dde4c033] Metal v1.4.2
[d96e819e] Parameters v0.12.3
[91a5bcdd] Plots v1.40.8
[92933f4c] ProgressMeter v1.10.2
[295af30f] Revise v3.6.2
[2913bbd2] StatsBase v0.34.3
[9a3f8284] Random
```

I observe:

```julia
using Metal
N = 10_000_000
X0 = MtlArray(zeros(Float32, Int64(N) ));
Metal.randn!(X0)
sum(isnan, X0) # random number, seems always > 0
```

I hope this is not too difficult to fix.

best regards

Contributor guide

No contributing guide indexed for this repository

Assessment

This issue has not been assessed yet.

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.