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

1import logging 

2 

3import torch 

4 

5from .copy import copy_ 

6 

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

8 

9 

10def contiguous(inp, memory_format=torch.contiguous_format): 

11 logger.debug("GEMS_CAMBRICON CONTIGUOUS") 

12 assert memory_format == torch.contiguous_format 

13 if inp.is_contiguous(memory_format=memory_format): 

14 return inp 

15 out = torch.empty_like(inp, memory_format=memory_format) 

16 return copy_(out, inp)