SWDEV-240806 - Add methods to update kernel command parameters

Change-Id: Iba90a31f9c5d6d4f2b60b7ccf903325c03d4d245
2021-07-22 08:03:31 -07:00
parent de5168fdef
commit 050d54b503
2 changed files with 19 additions and 0 deletions
@@ -915,9 +915,18 @@ class NDRangeKernelCommand : public Command {
  //! Return the kernel NDRange.
  const NDRangeContainer& sizes() const { return sizes_; }

+  //! updates kernel NDRange.
+  void setSizes(const size_t* globalWorkOffset, const size_t* globalWorkSize,
+                const size_t* localWorkSize) {
+    sizes_.update(3, globalWorkOffset, globalWorkSize, localWorkSize);
+  }
+
  //! Return the shared memory size
  uint32_t sharedMemBytes() const { return sharedMemBytes_; }

+  //! updates shared memory size
+  uint32_t setSharedMemBytes(uint32_t sharedMemBytes) { sharedMemBytes_ = sharedMemBytes; }
+
  //! Return the cooperative groups mode
  bool cooperativeGroups() const { return (extraParam_ & CooperativeGroups) ? true : false; }