assert trivially copyable type

Author: iclementine

include/triton_jit/triton_kernel.h

@@ -13,12 +13,7 @@ class TritonJITFunction;
 
 template <typename T>
 T get_next_multiple_of(T pos, T step) {
-  if (pos % step == 0) return pos;
-
-  while (pos % step) {
-    pos++;
-  }
-  return pos;
+  return ((pos + step - 1) / step) * step;
 }
 
 struct ParameterBuffer {
@@ -34,6 +29,7 @@ struct ParameterBuffer {
 
   template <typename T>
   void push_arg(T &&v) {
+    static_assert(std::is_trivially_copyable_v<std::remove_reference_t<T>>, "Non trivially copyable type");
     size_t align = alignof(T);
     size_t offset = get_next_multiple_of(this->cursor_, align);
     this->offsets_.push_back(offset);