Fix potential shader compile errors with dot product node, make sure B is casted to A so they have the same dimension