partial fix for facebookresearch#516

xformers/triton/layer_norm.py

@@ -221,7 +221,7 @@ def layer_norm(
             and bias is not None
         ):
             return _LayerNorm.apply(x, weight, bias, eps)
-    except (triton.code_gen.OutOfResources, RuntimeError) as e:
+    except RuntimeError as e:
         # Catch cases where the current GPU does not have enough registers to hold a full tensor line
         # fallback to PyTorch's implementation, which streams the tensor in and out
         _triton_registered_warnings = True