ROCm/composable_kernel
1
0
Fork 0
mirror of https://github.com/ROCm/composable_kernel.git synced 2026-05-24 14:54:47 +00:00
Code Issues Packages Projects Releases Wiki Activity

Merge commit 'ad57f6ef0bcaeef7988bfd3954aac06554f12afb' into develop

Browse Source
This commit is contained in:
assistant-librarian[bot]
2025-11-19 11:12:09 +00:00
parent 05f83b643f
commit 055f643f4f
235 changed files with 964 additions and 22 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
example/34_batchnorm/batchnorm_backward_nhwc.cpp
View File

@@ -14,6 +14,10 @@
#include "ck/library/reference_tensor_operation/cpu/reference_batchnorm_backward.hpp"
#include "ck/tensor_operation/gpu/device/impl/device_batchnorm_backward_impl.hpp"
using ::ck::DeviceMem;
using ::ck::HostTensorDescriptor;
using ::ck::Tensor;
static struct option long_options[] = {{"inOutLengths", required_argument, nullptr, 'D'},
{"verify", required_argument, nullptr, 'v'},
{"help", no_argument, nullptr, '?'},

4
example/34_batchnorm/batchnorm_forward_inferring_nhwc.cpp
View File

@@ -20,6 +20,10 @@
#include "batchnorm_infer_impl.hpp"
using ::ck::DeviceMem;
using ::ck::HostTensorDescriptor;
using ::ck::Tensor;
static struct option long_options[] = {{"inOutLengths", required_argument, nullptr, 'D'},
{"verify", required_argument, nullptr, 'v'},
{"help", no_argument, nullptr, '?'},

4
example/34_batchnorm/batchnorm_forward_training_nhwc.cpp
View File

@@ -20,6 +20,10 @@
#include "ck/library/utility/host_common_util.hpp"
#include "ck/tensor_operation/gpu/element/element_wise_operation.hpp"
using ::ck::DeviceMem;
using ::ck::HostTensorDescriptor;
using ::ck::Tensor;
static struct option long_options[] = {{"inOutLengths", required_argument, nullptr, 'D'},
{"verify", required_argument, nullptr, 'v'},
{"help", no_argument, nullptr, '?'},

4
example/34_batchnorm/batchnorm_forward_training_nhwc_obsolete.cpp
View File

@@ -20,6 +20,10 @@
#include "ck/library/utility/host_common_util.hpp"
#include "ck/tensor_operation/gpu/element/element_wise_operation.hpp"
using ::ck::DeviceMem;
using ::ck::HostTensorDescriptor;
using ::ck::Tensor;
static struct option long_options[] = {{"inOutLengths", required_argument, nullptr, 'D'},
{"verify", required_argument, nullptr, 'v'},
{"help", no_argument, nullptr, '?'},