Commit 0ef4b190 authored by James Keeling's avatar James Keeling Committed by TensorFlower Gardener
Browse files

Split topk GPU code into multiple files 

This file was a bottleneck during compilation, often taking many minutes to compile. In local testing this change reduces the wall-clock build time for the topk GPU kernels from 155s to 48s.

PiperOrigin-RevId: 228302319
parent 81492c07
Show whitespace changes
Inline Side-by-side
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Please to comment