Skip to main content

cubecl_cpp/hip/
atomic.rs

1use cubecl_core::ir::{dialect::atomic::*, prelude::*};
2
3use crate::hip::hip_op_with_out;
4
5hip_op_with_out!(AtomicIAddOp, |op, ctx| {
6    let ptr = op.ptr(ctx).name(ctx);
7    let value = op.value(ctx).name(ctx);
8    format!("atomicAdd({ptr}, {value})")
9});
10
11hip_op_with_out!(AtomicFAddOp, |op, ctx| {
12    let ptr = op.ptr(ctx).name(ctx);
13    let value = op.value(ctx).name(ctx);
14    format!("atomicAdd({ptr}, {value})")
15});