Home
last modified time | relevance | path

Searched refs:max_test_kernel (Results 1 – 1 of 1) sorted by relevance

/external/pytorch/aten/src/ATen/test/
Dcuda_atomic_ops_test.cu30 __global__ void max_test_kernel(T * a, T * max) { in max_test_kernel() function
146 max_test_kernel<<<dimGrid, dimBlock>>>(ad, sumd); in test_atomic_max()