tests/testthat/test-matmult.R

In gputools: GPU Drop-Ins for Statistics on Wide Data

set.seed(1)
a <- matrix(rnorm(60 * 40), 60, 40)
b <- matrix(rnorm(40 * 30), 40, 30)

test_that("gpuMatMult matches %*%", {
  skip_if_no_gpu()
  expect_equal(gpuMatMult(a, b), a %*% b, tolerance = 1e-10)
})

test_that("gpuCrossprod and gpuTcrossprod match base", {
  skip_if_no_gpu()
  expect_equal(gpuCrossprod(a), crossprod(a), tolerance = 1e-10)
  expect_equal(gpuTcrossprod(b), tcrossprod(b), tolerance = 1e-10)
})

test_that("mismatched dimensions are an error", {
  expect_error(gpuMatMult(a, a), "mismatched")
})

test_that("integer matrices are multiplied as numbers", {
  skip_if_no_gpu()
  m <- matrix(1:6, 2, 3)
  expect_equal(gpuMatMult(m, t(m)), m %*% t(m))
})