Skip to content

Change reduce_value type from double to float in CudaKernelOps#6

Open
dkapur17 wants to merge 1 commit intollmsystem:mainfrom
dkapur17:main
Open

Change reduce_value type from double to float in CudaKernelOps#6
dkapur17 wants to merge 1 commit intollmsystem:mainfrom
dkapur17:main

Conversation

@dkapur17
Copy link
Copy Markdown

Fixes data type mismatch between the Python binding and CUDA code (which expects a float), potentially leading to nan gradients. See https://edstem.org/us/courses/89798/discussion/7693487

Fixes data type mismatch between the Python binding and CUDA code (which expects a float), potentially leading to nan gradients.
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant