submission 673503

# -*- coding: utf-8 -*-
import os
os.environ["CUBLAS_WORKSPACE_CONFIG"] = ":4096:8"

import torch

# 1. PRE-CONFIGURACIÓN (World Line Stabilization)
# Seteamos esto una sola vez a nivel de módulo para ganar ~30µs por llamada.
torch.use_deterministic_algorithms(True)

def custom_kernel(data):
    """
    Sentinel Omega - Scan Inclusivo Turbo.
    """
    # 2. Recta de Vértices
    # Accedemos directamente a data[0]. No necesitamos el buffer data[1] 
    # si el protocolo exige un tensor de salida nuevo en float64.
    x = data[0]
    
    # 3. Colapso de Fase (Precision Control)
    # Cast y suma en una sola ráfaga de GPU.
    # El paso a float64 es ley para evitar el numerical instability (sqrt(n)).
    return torch.cumsum(x.to(torch.float64), dim=0)