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

1import logging 

2 

3import torch 

4 

5from .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_CAMBRICON RESOLVE_NEG") 

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