Skip to content

GitLab

"plugins/vscode:/vscode.git/clone" did not exist on "2443dceee5967194712cb89b056270697711594b"
Commit 5980100d authored by Peter Eastman's avatar Peter Eastman
Browse files

Fixed performance bug in mixed precision

parent b0161939
Show whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
platforms/cuda/src/CudaNonbondedUtilities.cpp
View file @ 5980100d
...@@ -446,7 +446,7 @@ CUfunction CudaNonbondedUtilities::createInteractionKernel(const string& source, ...@@ -446,7 +446,7 @@ CUfunction CudaNonbondedUtilities::createInteractionKernel(const string& source,
defines["NUM_ATOMS"] = context.intToString(context.getNumAtoms()); defines["NUM_ATOMS"] = context.intToString(context.getNumAtoms());
defines["PADDED_NUM_ATOMS"] = context.intToString(context.getPaddedNumAtoms()); defines["PADDED_NUM_ATOMS"] = context.intToString(context.getPaddedNumAtoms());
defines["NUM_BLOCKS"] = context.intToString(context.getNumAtomBlocks()); defines["NUM_BLOCKS"] = context.intToString(context.getNumAtomBlocks());
if ((localDataSize/4)%2 == 0 && !context.getUseDoublePrecision() && !context.getUseMixedPrecision()) if ((localDataSize/4)%2 == 0 && !context.getUseDoublePrecision())
defines["PARAMETER_SIZE_IS_EVEN"] = "1"; defines["PARAMETER_SIZE_IS_EVEN"] = "1";
if (context.getComputeCapability() >= 3.0 && !context.getUseDoublePrecision()) if (context.getComputeCapability() >= 3.0 && !context.getUseDoublePrecision())
defines["ENABLE_SHUFFLE"] = "1"; defines["ENABLE_SHUFFLE"] = "1";
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment