Skip to content

GitLab

Commit a30c626b authored by Bartlomiej Wroblewski's avatar Bartlomiej Wroblewski
Browse files

Make ComputeDataType an optional argument

parent b019d839
Hide whitespace changes
Inline Side-by-side
Showing with 40 additions and 40 deletions
library/src/tensor_operation_instance/gpu/contraction_bilinear/device_contraction_bilinear_m2_n2_k2_xdl_c_shuffle_f32_f32_f32_f32_compute_f16_kknn_instance.cpp
View file @ a30c626b
......@@ -41,10 +41,10 @@ void add_device_contraction_bilinear_m2_n2_k2_xdl_c_shuffle_f32_f32_f32_f32_comp
F32,
F32_Tuple,
F32,
F16,
PassThrough,
PassThrough,
Bilinear>>>& instances)
Bilinear,
F16>>>& instances)
{
add_device_operation_instances(
instances,
......
library/src/tensor_operation_instance/gpu/contraction_bilinear/device_contraction_bilinear_m2_n2_k2_xdl_c_shuffle_f32_f32_f32_f32_compute_f16_knnn_instance.cpp
View file @ a30c626b
......@@ -41,10 +41,10 @@ void add_device_contraction_bilinear_m2_n2_k2_xdl_c_shuffle_f32_f32_f32_f32_comp
F32,
F32_Tuple,
F32,
F16,
PassThrough,
PassThrough,
Bilinear>>>& instances)
Bilinear,
F16>>>& instances)
{
add_device_operation_instances(
instances,
......
library/src/tensor_operation_instance/gpu/contraction_bilinear/device_contraction_bilinear_m2_n2_k2_xdl_c_shuffle_f32_f32_f32_f32_compute_f16_mknn_instance.cpp
View file @ a30c626b
......@@ -41,10 +41,10 @@ void add_device_contraction_bilinear_m2_n2_k2_xdl_c_shuffle_f32_f32_f32_f32_comp
F32,
F32_Tuple,
F32,
F16,
PassThrough,
PassThrough,
Bilinear>>>& instances)
Bilinear,
F16>>>& instances)
{
add_device_operation_instances(
instances,
......
library/src/tensor_operation_instance/gpu/contraction_bilinear/device_contraction_bilinear_m2_n2_k2_xdl_c_shuffle_f32_f32_f32_f32_compute_f16_mnnn_instance.cpp
View file @ a30c626b
......@@ -41,10 +41,10 @@ void add_device_contraction_bilinear_m2_n2_k2_xdl_c_shuffle_f32_f32_f32_f32_comp
F32,
F32_Tuple,
F32,
F16,
PassThrough,
PassThrough,
Bilinear>>>& instances)
Bilinear,
F16>>>& instances)
{
add_device_operation_instances(
instances,
......
library/src/tensor_operation_instance/gpu/contraction_bilinear/device_contraction_bilinear_m2_n2_k2_xdl_c_shuffle_f32_f32_f32_f32_kknn_instance.cpp
View file @ a30c626b
......@@ -41,10 +41,10 @@ void add_device_contraction_bilinear_m2_n2_k2_xdl_c_shuffle_f32_f32_f32_f32_kknn
F32,
F32_Tuple,
F32,
F32,
PassThrough,
PassThrough,
Bilinear>>>& instances)
Bilinear,
F32>>>& instances)
{
add_device_operation_instances(
instances,
......
library/src/tensor_operation_instance/gpu/contraction_bilinear/device_contraction_bilinear_m2_n2_k2_xdl_c_shuffle_f32_f32_f32_f32_knnn_instance.cpp
View file @ a30c626b
......@@ -41,10 +41,10 @@ void add_device_contraction_bilinear_m2_n2_k2_xdl_c_shuffle_f32_f32_f32_f32_knnn
F32,
F32_Tuple,
F32,
F32,
PassThrough,
PassThrough,
Bilinear>>>& instances)
Bilinear,
F32>>>& instances)
{
add_device_operation_instances(
instances,
......
library/src/tensor_operation_instance/gpu/contraction_bilinear/device_contraction_bilinear_m2_n2_k2_xdl_c_shuffle_f32_f32_f32_f32_mknn_instance.cpp
View file @ a30c626b
......@@ -41,10 +41,10 @@ void add_device_contraction_bilinear_m2_n2_k2_xdl_c_shuffle_f32_f32_f32_f32_mknn
F32,
F32_Tuple,
F32,
F32,
PassThrough,
PassThrough,
Bilinear>>>& instances)
Bilinear,
F32>>>& instances)
{
add_device_operation_instances(
instances,
......
library/src/tensor_operation_instance/gpu/contraction_bilinear/device_contraction_bilinear_m2_n2_k2_xdl_c_shuffle_f32_f32_f32_f32_mnnn_instance.cpp
View file @ a30c626b
......@@ -41,10 +41,10 @@ void add_device_contraction_bilinear_m2_n2_k2_xdl_c_shuffle_f32_f32_f32_f32_mnnn
F32,
F32_Tuple,
F32,
F32,
PassThrough,
PassThrough,
Bilinear>>>& instances)
Bilinear,
F32>>>& instances)
{
add_device_operation_instances(
instances,
......
library/src/tensor_operation_instance/gpu/contraction_bilinear/device_contraction_bilinear_m2_n2_k2_xdl_c_shuffle_f64_f64_f64_f64_compute_f32_kknn_instance.cpp
View file @ a30c626b
......@@ -41,10 +41,10 @@ void add_device_contraction_bilinear_m2_n2_k2_xdl_c_shuffle_f64_f64_f64_f64_comp
F64,
F64_Tuple,
F64,
F32,
PassThrough,
PassThrough,
Bilinear>>>& instances)
Bilinear,
F32>>>& instances)
{
add_device_operation_instances(
instances,
......
library/src/tensor_operation_instance/gpu/contraction_bilinear/device_contraction_bilinear_m2_n2_k2_xdl_c_shuffle_f64_f64_f64_f64_compute_f32_knnn_instance.cpp
View file @ a30c626b
......@@ -41,10 +41,10 @@ void add_device_contraction_bilinear_m2_n2_k2_xdl_c_shuffle_f64_f64_f64_f64_comp
F64,
F64_Tuple,
F64,
F32,
PassThrough,
PassThrough,
Bilinear>>>& instances)
Bilinear,
F32>>>& instances)
{
add_device_operation_instances(
instances,
......
library/src/tensor_operation_instance/gpu/contraction_bilinear/device_contraction_bilinear_m2_n2_k2_xdl_c_shuffle_f64_f64_f64_f64_compute_f32_mknn_instance.cpp
View file @ a30c626b
......@@ -41,10 +41,10 @@ void add_device_contraction_bilinear_m2_n2_k2_xdl_c_shuffle_f64_f64_f64_f64_comp
F64,
F64_Tuple,
F64,
F32,
PassThrough,
PassThrough,
Bilinear>>>& instances)
Bilinear,
F32>>>& instances)
{
add_device_operation_instances(
instances,
......
library/src/tensor_operation_instance/gpu/contraction_bilinear/device_contraction_bilinear_m2_n2_k2_xdl_c_shuffle_f64_f64_f64_f64_compute_f32_mnnn_instance.cpp
View file @ a30c626b
......@@ -41,10 +41,10 @@ void add_device_contraction_bilinear_m2_n2_k2_xdl_c_shuffle_f64_f64_f64_f64_comp
F64,
F64_Tuple,
F64,
F32,
PassThrough,
PassThrough,
Bilinear>>>& instances)
Bilinear,
F32>>>& instances)
{
add_device_operation_instances(
instances,
......
library/src/tensor_operation_instance/gpu/contraction_bilinear/device_contraction_bilinear_m2_n2_k2_xdl_c_shuffle_f64_f64_f64_f64_kknn_instance.cpp
View file @ a30c626b
......@@ -41,10 +41,10 @@ void add_device_contraction_bilinear_m2_n2_k2_xdl_c_shuffle_f64_f64_f64_f64_kknn
F64,
F64_Tuple,
F64,
F64,
PassThrough,
PassThrough,
Bilinear>>>& instances)
Bilinear,
F64>>>& instances)
{
add_device_operation_instances(
instances,
......
library/src/tensor_operation_instance/gpu/contraction_bilinear/device_contraction_bilinear_m2_n2_k2_xdl_c_shuffle_f64_f64_f64_f64_knnn_instance.cpp
View file @ a30c626b
......@@ -41,10 +41,10 @@ void add_device_contraction_bilinear_m2_n2_k2_xdl_c_shuffle_f64_f64_f64_f64_knnn
F64,
F64_Tuple,
F64,
F64,
PassThrough,
PassThrough,
Bilinear>>>& instances)
Bilinear,
F64>>>& instances)
{
add_device_operation_instances(
instances,
......
library/src/tensor_operation_instance/gpu/contraction_bilinear/device_contraction_bilinear_m2_n2_k2_xdl_c_shuffle_f64_f64_f64_f64_mknn_instance.cpp
View file @ a30c626b
......@@ -41,10 +41,10 @@ void add_device_contraction_bilinear_m2_n2_k2_xdl_c_shuffle_f64_f64_f64_f64_mknn
F64,
F64_Tuple,
F64,
F64,
PassThrough,
PassThrough,
Bilinear>>>& instances)
Bilinear,
F64>>>& instances)
{
add_device_operation_instances(
instances,
......
library/src/tensor_operation_instance/gpu/contraction_bilinear/device_contraction_bilinear_m2_n2_k2_xdl_c_shuffle_f64_f64_f64_f64_mnnn_instance.cpp
View file @ a30c626b
......@@ -41,10 +41,10 @@ void add_device_contraction_bilinear_m2_n2_k2_xdl_c_shuffle_f64_f64_f64_f64_mnnn
F64,
F64_Tuple,
F64,
F64,
PassThrough,
PassThrough,
Bilinear>>>& instances)
Bilinear,
F64>>>& instances)
{
add_device_operation_instances(
instances,
......
library/src/tensor_operation_instance/gpu/contraction_scale/device_contraction_scale_m2_n2_k2_xdl_c_shuffle_bf16_bf16_bf16_compute_f32_kkn_instance.cpp
View file @ a30c626b
......@@ -41,10 +41,10 @@ void add_device_contraction_scale_m2_n2_k2_xdl_c_shuffle_bf16_bf16_bf16_compute_
BF16,
Empty_Tuple,
BF16,
F32,
PassThrough,
PassThrough,
Scale>>>& instances)
Scale,
F32>>>& instances)
{
add_device_operation_instances(
instances,
......
library/src/tensor_operation_instance/gpu/contraction_scale/device_contraction_scale_m2_n2_k2_xdl_c_shuffle_bf16_bf16_bf16_compute_f32_knn_instance.cpp
View file @ a30c626b
......@@ -41,10 +41,10 @@ void add_device_contraction_scale_m2_n2_k2_xdl_c_shuffle_bf16_bf16_bf16_compute_
BF16,
Empty_Tuple,
BF16,
F32,
PassThrough,
PassThrough,
Scale>>>& instances)
Scale,
F32>>>& instances)
{
add_device_operation_instances(
instances,
......
library/src/tensor_operation_instance/gpu/contraction_scale/device_contraction_scale_m2_n2_k2_xdl_c_shuffle_bf16_bf16_bf16_compute_f32_mkn_instance.cpp
View file @ a30c626b
......@@ -41,10 +41,10 @@ void add_device_contraction_scale_m2_n2_k2_xdl_c_shuffle_bf16_bf16_bf16_compute_
BF16,
Empty_Tuple,
BF16,
F32,
PassThrough,
PassThrough,
Scale>>>& instances)
Scale,
F32>>>& instances)
{
add_device_operation_instances(
instances,
......
library/src/tensor_operation_instance/gpu/contraction_scale/device_contraction_scale_m2_n2_k2_xdl_c_shuffle_bf16_bf16_bf16_compute_f32_mnn_instance.cpp
View file @ a30c626b
......@@ -41,10 +41,10 @@ void add_device_contraction_scale_m2_n2_k2_xdl_c_shuffle_bf16_bf16_bf16_compute_
BF16,
Empty_Tuple,
BF16,
F32,
PassThrough,
PassThrough,
Scale>>>& instances)
Scale,
F32>>>& instances)
{
add_device_operation_instances(
instances,
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment