Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dir_emitter.cc327 llvm::Value* atomic_memory_address; in EmitAtomicOperationUsingCAS() local
336 atomic_memory_address = PtrToInt(output_address, address_int_type); in EmitAtomicOperationUsingCAS()
338 llvm::Value* offset = And(atomic_memory_address, mask); in EmitAtomicOperationUsingCAS()
340 atomic_memory_address = And(atomic_memory_address, mask); in EmitAtomicOperationUsingCAS()
341 atomic_memory_address = in EmitAtomicOperationUsingCAS()
342 IntToPtr(atomic_memory_address, atomic_address_type); in EmitAtomicOperationUsingCAS()
347 atomic_memory_address = BitCast(output_address, atomic_address_type); in EmitAtomicOperationUsingCAS()
354 llvm::Value* cas_old_output = Load(atomic_memory_address, "cas_old_output"); in EmitAtomicOperationUsingCAS()
381 AtomicCmpXchg(atomic_memory_address, cas_old_output, cas_new_output, in EmitAtomicOperationUsingCAS()