Coverage for src/flag_gems/runtime/backend/_kunlunxin/ops/resolve_neg.py: 0%

1import logging 

2 

3import torch 

4 

5from flag_gems.ops.neg import neg_func 

6 

7logger = logging.getLogger("flag_gems").getChild(__name__.lstrip(".")) 

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