Merge pull request #5165 from halide/abadams/rungen_set_host_dirty

abadams · web-flow · commit 52da814a2c3c · 2020-08-08T17:46:32.000-07:00
Make sure to set_host_dirty in rungen
diff --git a/tools/RunGen.h b/tools/RunGen.h
@@ -387,6 +387,7 @@ inline Buffer<> allocate_buffer(const halide_type_t &type, const Shape &shape) {
     if (b.number_of_elements() > 0) {
         b.check_overflow();
         b.allocate();
+        b.set_host_dirty();
     }
     return b;
 }
@@ -875,6 +876,10 @@ struct ArgData {
 
         buffer_value = allocate_buffer(metadata->type, new_shape);
 
+        // allocate_buffer conservatively sets host dirty. Don't waste
+        // time copying output buffers to device.
+        buffer_value.set_host_dirty(false);
+
         info() << "Output " << name << ": BoundsQuery result is " << constrained_shape;
         info() << "Output " << name << ": Shape is " << get_shape(buffer_value);
     }

Original file line number	Diff line number	Diff line change
`@@ -387,6 +387,7 @@ inline Buffer<> allocate_buffer(const halide_type_t &type, const Shape &shape) {`
`387`	`387`	`if (b.number_of_elements() > 0) {`
`388`	`388`	`b.check_overflow();`
`389`	`389`	`b.allocate();`
	`390`	`+ b.set_host_dirty();`
`390`	`391`	`}`
`391`	`392`	`return b;`
`392`	`393`	`}`
`@@ -875,6 +876,10 @@ struct ArgData {`
`875`	`876`
`876`	`877`	`buffer_value = allocate_buffer(metadata->type, new_shape);`
`877`	`878`
	`879`	`+ // allocate_buffer conservatively sets host dirty. Don't waste`
	`880`	`+ // time copying output buffers to device.`
	`881`	`+ buffer_value.set_host_dirty(false);`
	`882`	`+`
`878`	`883`	`info() << "Output " << name << ": BoundsQuery result is " << constrained_shape;`
`879`	`884`	`info() << "Output " << name << ": Shape is " << get_shape(buffer_value);`
`880`	`885`	`}`