Coverage for src/flag_gems/ops/resolve_neg.py: 100%

1import logging 

2 

3import torch 

4 

5from flag_gems.ops.neg import neg_func 

6 

7logger = logging.getLogger(__name__) 

8 

9 

10def resolve_neg(A: torch.Tensor): 

11 logger.debug("GEMS RESOLVE_NEG") 

12 return neg_func(A) if A.is_neg() else A