X-Git-Url: https://git.kernel.dk/?p=fio.git;a=blobdiff_plain;f=profile.c;h=506462eb85373d795f745b0a4e479bd9a1937f1e;hp=0e2b97d4a0933896532b4891bd650dcf63c16da0;hb=d72be5454c8c5378f16804ff9b8d1afe8729a380;hpb=15dc1934435dc84d66547c4fc92d936224d7238f;ds=sidebyside

diff --git a/profile.c b/profile.c
index 0e2b97d4..506462eb 100644
--- a/profile.c
+++ b/profile.c
@@ -93,6 +93,26 @@ void profile_add_hooks(struct thread_data *td)
 	if (!ops)
 		return;
 
-	td->fill_io_u_off = ops->fill_io_u_off;
-	td->fill_io_u_size = ops->fill_io_u_size;
+	if (ops->io_ops) {
+		td->prof_io_ops = *ops->io_ops;
+		td->flags |= TD_F_PROFILE_OPS;
+	}
+}
+
+int profile_td_init(struct thread_data *td)
+{
+	struct prof_io_ops *ops = &td->prof_io_ops;
+
+	if (ops->td_init)
+		return ops->td_init(td);
+
+	return 0;
+}
+
+void profile_td_exit(struct thread_data *td)
+{
+	struct prof_io_ops *ops = &td->prof_io_ops;
+
+	if (ops->td_exit)
+		ops->td_exit(td);
 }