leavelet/cutlass
1
0
Fork 0
Code Issues Pull Requests Actions 28 Packages Projects Releases Wiki Activity

releaase 2.11 (#703)

Browse Source
This commit is contained in:
Aditya Atluri
2022-11-19 09:02:15 -05:00
committed by GitHub
parent 3c90f6aea6
commit c975e2ccbb
329 changed files with 47332 additions and 10607 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tools/library/src/gemm_operation.h
+15 -15
View File
@@ -305,7 +305,7 @@ public:
Operator *op = static_cast<Operator *>(host_workspace);
status = op->update(args, device_workspace);
status = op->update(args);
if (status != Status::kSuccess) {
return status;
@@ -507,7 +507,7 @@ public:
Operator *op = static_cast<Operator *>(host_workspace);
status = op->update(args, device_workspace);
status = op->update(args);
if (status != Status::kSuccess) {
return status;
@@ -725,8 +725,8 @@ public:
}
Operator *op = static_cast<Operator *>(host_workspace);
status = op->update(args, device_workspace);
status = op->update(args);
if (status != Status::kSuccess) {
return status;
@@ -917,30 +917,30 @@ public:
/// Runs the kernel
virtual Status run(
void const *arguments_ptr,
void *host_workspace,
void *device_workspace = nullptr,
void *host_workspace,
void *device_workspace = nullptr,
cudaStream_t stream = nullptr) const {
OperatorArguments args;
Status status = update_arguments_(
args,
args,
static_cast<GemmPlanarComplexArguments const *>(arguments_ptr));
if (status != Status::kSuccess) {
return status;
}
Operator *op = static_cast<Operator *>(host_workspace);
status = op->update(args, device_workspace);
status = op->update(args);
if (status != Status::kSuccess) {
return status;
}
status = op->run(stream);
return status;
}
};
@@ -1134,7 +1134,7 @@ public:
Operator *op = static_cast<Operator *>(host_workspace);
status = op->update(args, device_workspace);
status = op->update(args);
if (status != Status::kSuccess) {
return status;
@@ -1336,7 +1336,7 @@ public:
Operator *op = static_cast<Operator *>(host_workspace);
status = op->update(args, device_workspace);
status = op->update(args);
if (status != Status::kSuccess) {
return status;