submission 545032

Use it

Vendorable · source mirrored · license unknownView source →

Benchmark evidence

1 measurement across 1 GPU, fastest first.

Reported · How evidence levels are derived →

Source and license

Kernel source

import os
os.environ["CUBLAS_WORKSPACE_CONFIG"] = ":4096:8"

import torch
from task import input_t, output_t

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

Source code from GPU Mode and the KernelBot dataset · June 9 Researcher Reciprocity License v1.0

Changes from previous submission

Against this author's previous submission submission 544990.

  ⋯ 5 unchanged lines
   
  def custom_kernel(data: input_t) -> output_t:
      data, output = data
-     torch.sort(data, out=(output, torch.empty_like(data, dtype=torch.long)))
+     output[...] = torch.sort(data)[0]
      return output