Skip to content

Commit 4a404b4

Browse files
author
UESTC-AHao
committed
fixed
1 parent a1c334b commit 4a404b4

File tree

3 files changed

+4
-5
lines changed

3 files changed

+4
-5
lines changed

ucm/store/device/cuda/cuda_device.cc

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -162,7 +162,7 @@ class CudaDevice : public IBufferedDevice {
162162
CUfileHandle_t cuFileHandle = nullptr;
163163
auto status = CuFileHandleRecorder::Instance().Get(path, cuFileHandle,
164164
[&path](CUfileHandle_t& handle, int& fd) -> Status {
165-
return CreateCuFileHandle(path, O_WRONLY | O_CREAT | O_DIRECT, handle, fd);
165+
return CreateCuFileHandle(path, O_WRONLY | O_DIRECT, handle, fd);
166166
});
167167
if (status.Failure()) {
168168
return status;

ucm/store/nfsstore/cc/domain/trans/directstorage_queue.cc

Lines changed: 0 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -78,9 +78,6 @@ void DirectStorageQueue::Done(Task::Shard& shard, const Device& device, const bo
7878
{
7979
if (!success) { this->failureSet_->Insert(shard.owner); }
8080
if (!shard.done) { return; }
81-
if (device) {
82-
if (device->Synchronized().Failure()) { this->failureSet_->Insert(shard.owner); }
83-
}
8481
shard.done();
8582
}
8683

ucm/store/task/task_queue.h

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -25,6 +25,8 @@
2525
#define UNIFIEDCACHE_TASK_QUEUE_H
2626

2727
#include "task_shard.h"
28+
#include "task_set.h"
29+
#include "../nfsstore/cc/domain/space/space_layout.h"
2830

2931
namespace UC {
3032

@@ -33,7 +35,7 @@ class TaskQueue {
3335
virtual ~TaskQueue() = default;
3436
virtual void Push(std::list<Task::Shard>& shards) noexcept = 0;
3537
virtual Status Setup(const int32_t deviceId, const size_t bufferSize, const size_t bufferNumber,
36-
class TaskSet* failureSet, const class SpaceLayout* layout, const size_t timeoutMs, bool transferUseDirect) = 0;
38+
TaskSet* failureSet, const SpaceLayout* layout, const size_t timeoutMs, bool transferUseDirect) = 0;
3739
};
3840

3941
} // namespace UC

0 commit comments

Comments
 (0)