Hello, I am looking for the way to compute cross product on a GPU. Basically something similar to this: function cross(w::CuArray{Float32,1}, u::CuArray{Float32,1}, v::CuArray{Float32,1}) w[1] = u[2] * v[3] - u[3] * v[2] w[2] = u[3] * v[1] - u[1] * v[3] w[3] = u[1] * v[2] - u[2] * v[1] end Is…

Cross product with CUDA.jl

maxfreu June 11, 2021, 4:31pm 3

If you need many cross products I‘d make two CuVectors containing many StaticArrays and then broadcast the cross product over them. Didn‘t think too much about it, but should work.

4 Likes

Topic		Replies	Views
ArrayFire Julia: Performant way to calculate vector cross product? GPU question , arrayfire	0	1429	January 13, 2017
Elegant kernel for vectors in matrix GPU cuda	3	508	February 5, 2021
Performance of view with cuArrays GPU	11	2708	November 11, 2018
Using vectors of vectors with CuArrays GPU	2	1551	July 9, 2019
Julia Cuda Matrix multiplication General Usage cudanative , cuda	3	4315	February 24, 2021

Cross product with CUDA.jl

Related topics