Bug fix.

mzient · mzient · commit e32b71dc5888 · 2022-02-10T18:33:08.000+01:00
diff --git a/dali/kernels/dynamic_scratchpad.h b/dali/kernels/dynamic_scratchpad.h
@@ -60,6 +60,8 @@ class DynamicScratchpadImplT {
                              AccessOrder dealloc_order = {}) {
     static_assert(!std::is_same<Kind, mm::memory_kind::host>::value,
       "Cannot use a stream-ordered resource for plain host memory");
+    if (dealloc_order.has_value())
+      dealloc_order = alloc_order;
     adapter<Kind>() = { rsrc, alloc_order, dealloc_order };
     set_upstream_resource<Kind>(&adapter<Kind>());
   }
diff --git a/dali/operators/image/resize/resize_base.cc b/dali/operators/image/resize/resize_base.cc
@@ -121,8 +121,8 @@ void ResizeBase<GPUBackend>::InitializeGPU(int minibatch_size, size_t temp_buffe
     minibatch_size_ = minibatch_size;
   }
   kmgr_.Resize(1, 0);
-  kmgr_.SetMemoryHint<mm::memory_kind::device>(temp_buffer_hint);
-  kmgr_.GetScratchpadAllocator(0).Reserve<mm::memory_kind::device>(temp_buffer_hint);
+  // kmgr_.SetMemoryHint<mm::memory_kind::device>(temp_buffer_hint);
+  // kmgr_.GetScratchpadAllocator(0).Reserve<mm::memory_kind::device>(temp_buffer_hint);
 }
 
 template <typename Backend>

Original file line number	Diff line number	Diff line change
`@@ -60,6 +60,8 @@ class DynamicScratchpadImplT {`
`60`	`60`	`AccessOrder dealloc_order = {}) {`
`61`	`61`	`static_assert(!std::is_same<Kind, mm::memory_kind::host>::value,`
`62`	`62`	`"Cannot use a stream-ordered resource for plain host memory");`
	`63`	`+ if (dealloc_order.has_value())`
	`64`	`+ dealloc_order = alloc_order;`
`63`	`65`	`adapter<Kind>() = { rsrc, alloc_order, dealloc_order };`
`64`	`66`	`set_upstream_resource<Kind>(&adapter<Kind>());`
`65`	`67`	`}`
Original file line number	Diff line number	Diff line change
`@@ -121,8 +121,8 @@ void ResizeBase<GPUBackend>::InitializeGPU(int minibatch_size, size_t temp_buffe`
`121`	`121`	`minibatch_size_ = minibatch_size;`
`122`	`122`	`}`
`123`	`123`	`kmgr_.Resize(1, 0);`
`124`		`- kmgr_.SetMemoryHint<mm::memory_kind::device>(temp_buffer_hint);`
`125`		`- kmgr_.GetScratchpadAllocator(0).Reserve<mm::memory_kind::device>(temp_buffer_hint);`
	`124`	`+ // kmgr_.SetMemoryHint<mm::memory_kind::device>(temp_buffer_hint);`
	`125`	`+ // kmgr_.GetScratchpadAllocator(0).Reserve<mm::memory_kind::device>(temp_buffer_hint);`
`126`	`126`	`}`
`127`	`127`
`128`	`128`	`template <typename Backend>`