fix cuda misalign

tensorflow · Apr 20, 2024 · eaf7911 · eaf7911
1 parent 60c9997
commit eaf7911
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/...flow_recommenders_addons/dynamic_embedding/core/kernels/lookup_impl/lookup_table_op_hkv.h b/...flow_recommenders_addons/dynamic_embedding/core/kernels/lookup_impl/lookup_table_op_hkv.h
@@ -332,7 +332,7 @@ class TFOrDefaultAllocator : public nv::merlin::BaseAllocator {
   std::unique_ptr<nv::merlin::DefaultAllocator> default_allocator_ = nullptr;
   bool use_default_allocator_ = false;
   // bool tf_async_allocator_stream_set_ = false;
-  static constexpr size_t kAllocatorAlignment = 8;
+  static constexpr size_t kAllocatorAlignment = 4;
 
  public:
   TFOrDefaultAllocator() : use_default_allocator_(true) {