Coverage for src/flag_gems/runtime/backend/_metax/ops/resolve_conj.py: 0%

1import logging 

2 

3import torch 

4 

5from flag_gems.ops.neg import neg_func 

6 

7logger = logging.getLogger("flag_gems." + __name__) 

8 

9 

10def resolve_conj(A: torch.Tensor): 

11 logger.debug("METAX GEMS RESOLVE_CONJ") 

12 return torch.complex(A.real, neg_func(A.imag)) if A.is_conj() else A