-
Notifications
You must be signed in to change notification settings - Fork 567
wgan example failed to run on H100 with cuda backend #3079
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
Comments
hey, in which branch are you getting this error? |
Thanks. I just use the main branch. |
Same error on 0.17.0 |
Thanks for the bug report! Will take a look. /edit: this is clearly a fusion bug
If you disable the default features (which includes fusion) it should work. [dependencies]
-burn = { path = "../../crates/burn", features=["train", "vision"] }
+burn = { path = "../../crates/burn", features=["train", "vision", "std"], default-features = false }
image = { workspace = true } Still a bug though 😅 |
Looks like we maxed out the parameter space with element-wise fusion. elemwise_fuse def
CC'ing @nathanielsimard |
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
0.17.0 version
H100 with CUDA 12.1
The text was updated successfully, but these errors were encountered: