X-Git-Url: http://plrg.eecs.uci.edu/git/?a=blobdiff_plain;ds=sidebyside;f=cmodelint.cc;h=47aef05f28a1c61e72101e171e59da02354c5f09;hb=4e4fa53414eb5f0a78e05478127b375d5f0aba73;hp=7e52d10798ffc8777ebe2d7d4915055300b32c4d;hpb=e60d8c23d30a0dfe66b8426f7f2ecf576e812028;p=c11tester.git

diff --git a/cmodelint.cc b/cmodelint.cc
index 7e52d107..47aef05f 100644
--- a/cmodelint.cc
+++ b/cmodelint.cc
@@ -1,11 +1,11 @@
 #include "model.h"
+#include "action.h"
 #include "cmodelint.h"
 #include "threads-model.h"
 
 /** Performs a read action.*/
 uint64_t model_read_action(void * obj, memory_order ord) {
-	model->switch_to_master(new ModelAction(ATOMIC_READ, ord, obj));
-	return thread_current()->get_return_value();
+	return model->switch_to_master(new ModelAction(ATOMIC_READ, ord, obj));
 }
 
 /** Performs a write action.*/
@@ -24,8 +24,7 @@ void model_init_action(void * obj, uint64_t val) {
  * a write.
  */
 uint64_t model_rmwr_action(void *obj, memory_order ord) {
-	model->switch_to_master(new ModelAction(ATOMIC_RMWR, ord, obj));
-	return thread_current()->get_return_value();
+	return model->switch_to_master(new ModelAction(ATOMIC_RMWR, ord, obj));
 }
 
 /** Performs the write part of a RMW action. */