submission 512776

# submission_sort_v2.py
# Sort a 1D float32 tensor in ascending order.
# Interface: custom_kernel((input_tensor, output_tensor)) -> output
#   input_tensor:  (size,) float32 on CUDA
#   output_tensor: (size,) float32 pre-allocated
#
# Strategy: delegate to torch.sort (cuDNN/CUB radix sort internally).
# A pure-Triton radix sort requires a stable scatter phase that is complex
# to implement correctly; torch.sort provides a correct, highly-optimised
# baseline.  Replacing with a custom Triton radix sort is a future step.

import torch
from task import input_t, output_t


def custom_kernel(data: input_t) -> output_t:
    x, out = data
    out.copy_(torch.sort(x)[0])
    return out