Coverage for src/flag_gems/ops/contiguous.py: 45%

1import logging 

2 

3import torch 

4 

5from flag_gems.ops.copy import copy_ 

6 

7logger = logging.getLogger(__name__) 

8 

9 

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

11 assert memory_format == torch.contiguous_format 

12 logger.debug("GEMS CONTIGUOUS") 

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)