We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
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
init_expectations
The initializer for expectations on the GPU is doing scalar indexing on https://github.com/Juice-jl/ProbabilisticCircuits.jl/blob/fc0c3acb9f0eb54fc1a12baf62d0c427a265a16f/src/queries/expectation_bit.jl#L88 This ends up happening on the GPU if ParamBitCircuitPair is on the GPU; scalar indexing on the GPU is very slow and a bad idea! We should move the BitCircuit to CPU or keep it there when doing these scalar operations. Perhaps postpone moving BitCircuit to GPU until after the initializer is called.
ParamBitCircuitPair
BitCircuit
The text was updated successfully, but these errors were encountered:
Wrote a cuda kernel to fix this 14fbf66
Sorry, something went wrong.
khosravipasha
No branches or pull requests
The initializer for expectations on the GPU is doing scalar indexing on https://github.com/Juice-jl/ProbabilisticCircuits.jl/blob/fc0c3acb9f0eb54fc1a12baf62d0c427a265a16f/src/queries/expectation_bit.jl#L88
This ends up happening on the GPU if
ParamBitCircuitPair
is on the GPU; scalar indexing on the GPU is very slow and a bad idea! We should move theBitCircuit
to CPU or keep it there when doing these scalar operations. Perhaps postpone movingBitCircuit
to GPU until after the initializer is called.The text was updated successfully, but these errors were encountered: