warp.atomic_max#

warp.atomic_max(arr: Array[Any], i: Int, value: Any) → Any#
  • Kernel
  • Differentiable

Compute the maximum of value and arr[i], atomically update the array, and return the old value.

The operation is only atomic on a per-component basis for vectors and matrices.

warp.atomic_max(arr: Array[Any], i: Int, j: Int, value: Any) → Any
  • Kernel
  • Differentiable

Compute the maximum of value and arr[i,j], atomically update the array, and return the old value.

The operation is only atomic on a per-component basis for vectors and matrices.

warp.atomic_max(arr: Array[Any], i: Int, j: Int, k: Int, value: Any) → Any
  • Kernel
  • Differentiable

Compute the maximum of value and arr[i,j,k], atomically update the array, and return the old value.

The operation is only atomic on a per-component basis for vectors and matrices.

warp.atomic_max(
arr: Array[Any],
i: Int,
j: Int,
k: Int,
l: Int,
value: Any,
) → Any
  • Kernel
  • Differentiable

Compute the maximum of value and arr[i,j,k,l], atomically update the array, and return the old value.

The operation is only atomic on a per-component basis for vectors and matrices.

warp.atomic_max(arr: FabricArray[Any], i: Int, value: Any) → Any
  • Kernel
  • Differentiable

Compute the maximum of value and arr[i], atomically update the array, and return the old value.

The operation is only atomic on a per-component basis for vectors and matrices.

warp.atomic_max(arr: FabricArray[Any], i: Int, j: Int, value: Any) → Any
  • Kernel
  • Differentiable

Compute the maximum of value and arr[i,j], atomically update the array, and return the old value.

The operation is only atomic on a per-component basis for vectors and matrices.

warp.atomic_max(
arr: FabricArray[Any],
i: Int,
j: Int,
k: Int,
value: Any,
) → Any
  • Kernel
  • Differentiable

Compute the maximum of value and arr[i,j,k], atomically update the array, and return the old value.

The operation is only atomic on a per-component basis for vectors and matrices.

warp.atomic_max(
arr: FabricArray[Any],
i: Int,
j: Int,
k: Int,
l: Int,
value: Any,
) → Any
  • Kernel
  • Differentiable

Compute the maximum of value and arr[i,j,k,l], atomically update the array, and return the old value.

The operation is only atomic on a per-component basis for vectors and matrices.

warp.atomic_max(arr: IndexedFabricArray[Any], i: Int, value: Any) → Any
  • Kernel
  • Differentiable

Compute the maximum of value and arr[i], atomically update the array, and return the old value.

The operation is only atomic on a per-component basis for vectors and matrices.

warp.atomic_max(
arr: IndexedFabricArray[Any],
i: Int,
j: Int,
value: Any,
) → Any
  • Kernel
  • Differentiable

Compute the maximum of value and arr[i,j], atomically update the array, and return the old value.

The operation is only atomic on a per-component basis for vectors and matrices.

warp.atomic_max(
arr: IndexedFabricArray[Any],
i: Int,
j: Int,
k: Int,
value: Any,
) → Any
  • Kernel
  • Differentiable

Compute the maximum of value and arr[i,j,k], atomically update the array, and return the old value.

The operation is only atomic on a per-component basis for vectors and matrices.

warp.atomic_max(
arr: IndexedFabricArray[Any],
i: Int,
j: Int,
k: Int,
l: Int,
value: Any,
) → Any
  • Kernel
  • Differentiable

Compute the maximum of value and arr[i,j,k,l], atomically update the array, and return the old value.

The operation is only atomic on a per-component basis for vectors and matrices.