BulkUpdateAsync taking too much time on tables with Partition

[Adnan-Niloy]

Here is my code:

  var bulkConfig = new BulkConfig
            {
                PropertiesToInclude = propertyNames,
                UpdateByProperties = ["Id", "BranchId"],
                UseTempDB = true,
                SetOutputIdentity = false,
            };
            await _context.BulkUpdateAsync(entities, bulkConfig);

Here Id is the primary key and BranchId is the partitioned column. I am trying to update 3 properties. When I try this on a table with no partition it works fine (I tried with 1000+ data with no issue). However, in this case, where there is a Partition on the table, I tried only with one row to update, but it took too long, resulting in an execution timeout from the db. Any thoughts?