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.cc418 llvm::Value* atomic_memory_address; in EmitAtomicOperationUsingCAS() local
427 atomic_memory_address = PtrToInt(output_address, address_int_type); in EmitAtomicOperationUsingCAS()
429 llvm::Value* offset = And(atomic_memory_address, mask); in EmitAtomicOperationUsingCAS()
431 atomic_memory_address = And(atomic_memory_address, mask); in EmitAtomicOperationUsingCAS()
432 atomic_memory_address = in EmitAtomicOperationUsingCAS()
433 IntToPtr(atomic_memory_address, atomic_address_type); in EmitAtomicOperationUsingCAS()
442 atomic_memory_address = b_.CreatePointerBitCastOrAddrSpaceCast( in EmitAtomicOperationUsingCAS()
453 llvm::Value* cas_old_output = Load(atomic_memory_address, "cas_old_output"); in EmitAtomicOperationUsingCAS()
480 AtomicCmpXchg(atomic_memory_address, cas_old_output, cas_new_output, in EmitAtomicOperationUsingCAS()