Remove hardcoded warp_size == 32 assumption from PME GPU

[alexxy/gromacs.git] / src / gromacs / ewald / pme-gpu-program.cpp

diff --git a/src/gromacs/ewald/pme-gpu-program.cpp b/src/gromacs/ewald/pme-gpu-program.cpp
index 50031f39ae82503a6283edbc24b0a4b84711f60e..46c2103e5e273cda0caf2324dc6c40d4e044efd7 100644 (file)
--- a/src/gromacs/ewald/pme-gpu-program.cpp
+++ b/src/gromacs/ewald/pme-gpu-program.cpp
@@ -60,5 +60,10 @@ PmeGpuProgram::~PmeGpuProgram() = default;
 
 PmeGpuProgramStorage buildPmeGpuProgram(const gmx_device_info_t *deviceInfo)
 {
+    if (!deviceInfo)
+    {
+        // This workaround is only needed for CodePath::CPU dummy in testhardwarecontexts.cpp
+        return nullptr;
+    }
     return gmx::compat::make_unique<PmeGpuProgram>(deviceInfo);
 }