Update bias_act.py

torch_utils/ops/bias_act.py

@@ -81,8 +81,8 @@ def bias_act(x, b=None, dim=1, act='linear', alpha=None, gain=None, clamp=None,
     """
     assert isinstance(x, torch.Tensor)
     assert impl in ['ref', 'cuda']
-    if impl == 'cuda' and x.device.type == 'cuda' and _init():
-        return _bias_act_cuda(dim=dim, act=act, alpha=alpha, gain=gain, clamp=clamp).apply(x, b)
+    # if impl == 'cuda' and x.device.type == 'cuda' and _init():
+    #     return _bias_act_cuda(dim=dim, act=act, alpha=alpha, gain=gain, clamp=clamp).apply(x, b)
     return _bias_act_ref(x=x, b=b, dim=dim, act=act, alpha=alpha, gain=gain, clamp=clamp)
 
 #----------------------------------------------------------------------------