-
-
Notifications
You must be signed in to change notification settings - Fork 30
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
Some automated conversions in saveat #309
Conversation
@utkarsh530 can you check what's going on with Metal? |
It should be ready now after the tests pass. |
DiffEqDevTools = "f3b72e0c-5b89-59e1-b016-84e28bfd966d" | ||
Distributed = "8ba89e20-285c-5b6f-9357-94700520ee1b" | ||
Flux = "587475ba-b771-5e3f-ad9e-33799f191a9c" | ||
ForwardDiff = "f6369f11-7733-5829-9624-2563aa707210" | ||
GPUArraysCore = "46192b85-c4d5-4398-a991-12ede77f4527" | ||
LinearAlgebra = "37e2e46d-f89d-539d-b4ee-838fcccc9c8e" | ||
Metal = "dde4c033-4e86-420c-a63e-0dd931031962" |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
is this intentional?
## Don't test the problems in which GPUs don't support FP64 completely yet | ||
## Creating StepRangeLen causes some param types to be FP64 inferred by `float` function |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
the conversion should change things to Float32?
@@ -0,0 +1,114 @@ | |||
|
|||
# saveat is just a bool here: |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
That's not allowed?
Merging so I can use it. But I'm still not sure about the bool dispatch. |
This makes it nicer to use Float32 from R and Python, since you can just
saveat=0.01
and it attempts to convert the type to Float32 if the JIT was to Float32