apex icon indicating copy to clipboard operation
apex copied to clipboard

Published 20 hours ago • verified publisher icon NVIDIA

`apex.contrib.group_norm` would better have an import guard of `group_norm_cuda`

Open crcrpar opened this issue 2 years ago • 1 comments

https://github.com/NVIDIA/apex/blob/50ac8425403b98147cbb66aea9a2a27dd3fe7673/apex/contrib/group_norm/group_norm.py#L21

Some contrib modules don't have one though e.g. https://github.com/NVIDIA/apex/blob/50ac8425403b98147cbb66aea9a2a27dd3fe7673/apex/contrib/layer_norm/layer_norm.py#L5

cc @ptrblck @xwang233

crcrpar avatar Jul 26 '23 20:07 crcrpar