Anirudh Ajith

icon indicating an email [email protected]

icon company Allen Institute for Artificial Intelligence icon location Seattle, Washington, USA icon location Predoctoral Researcher at Ai2

Results 2 issues of Anirudh Ajith

faiss::gpu::runMatrixMult ... cublas failed (13): (1024, 12) x (256, 12)' = (1024, 256) gemm params m 256 n 1024 k 12 trA T trB N lda 12 ldb 12 ldc 256

23 comment

# Summary I'm trying to train an IVFPQ index for 100000 768-dimensional embeddings on an NVIDIA GPU with 40537MiB of memory. The code fails at `index.train()` with the following error...

GPU

Variable tracking size exceeded error on running build.sh while compiling spike_main/disasm.cc

2 comment

I run Ubuntu 18.04. On a fresh clone of riscv-tools, I followed (as far as I can tell) the instructions in `README.md`. On running `./build.sh` as mentioned, I got the...