提交 82c3ed0d authored 作者: Arnaud Bergeron's avatar Arnaud Bergeron

Use the right mask to check for alignement.

上级 0d47e204
......@@ -257,7 +257,7 @@ static std::string shape(cudnnConvolutionDescriptor_t convDesc, int dataTypecode
static bool all_aligned(cudnnDataType_t type, void* in, void* out, void* filter)
{
size_t alignMask = (type == CUDNN_DATA_HALF) ? 0x7F : 0xFF ;
size_t alignMask = 0xF;
// there have to be entries for both aligned and not
if (((size_t)in | (size_t)out | (size_t)filter) & alignMask)
{
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论