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

1import logging 

2 

3import torch 

4 

5from .full import check_dtype, full_ 

6 

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

8 

9 

10def full_like( 

11 x, 

12 fill_value, 

13 *, 

14 dtype=None, 

15 layout=None, 

16 device=None, 

17 pin_memory=None, 

18 memory_format=None, 

19): 

20 logger.debug("METAX GEMS FULL_LIKE") 

21 if device is None: 

22 device = x.device 

23 if dtype is None: 

24 dtype = x.dtype 

25 fill_value = check_dtype(fill_value, dtype, device) 

26 out = torch.empty_like(x, device=device, dtype=dtype) 

27 N = x.numel() 

28 

29 return full_(out, N, dtype, device, fill_value)