Skip to content

Commit

Permalink
transpose points
Browse files Browse the repository at this point in the history
  • Loading branch information
mscroggs committed Jul 26, 2024
1 parent 1ebb89e commit db51197
Showing 1 changed file with 3 additions and 3 deletions.
6 changes: 3 additions & 3 deletions src/assembly/batched/potential.rs
Original file line number Diff line number Diff line change
Expand Up @@ -33,8 +33,8 @@ fn assemble_batch<
table: &RlstArray<T, 4>,
) -> usize {
let npts = weights.len();
let nevalpts = evaluation_points.shape()[0];
debug_assert!(points.shape()[0] == npts);
let nevalpts = evaluation_points.shape()[1];
debug_assert!(points.shape()[1] == npts);

let grid = space.grid();

Expand Down Expand Up @@ -167,7 +167,7 @@ pub trait BatchedPotentialAssembler: Sync + Sized {
if !space.is_serial() {
panic!("Dense assembly can only be used for function spaces stored in serial");
}
if output.shape()[0] != points.shape()[0] || output.shape()[1] != space.global_size() {
if output.shape()[0] != points.shape()[1] || output.shape()[1] != space.global_size() {
panic!("Matrix has wrong shape");
}

Expand Down

0 comments on commit db51197

Please sign in to comment.