saymrwulf/onnxruntime
1
0
Fork 0
mirror of https://github.com/saymrwulf/onnxruntime.git synced 2026-05-18 21:21:17 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Add license header to CUDA related files (#16437)

Browse source 
Add license header for files under core/providers/cuda or contrib_ops/cuda/
This commit is contained in:
Tianlei Wu 2023-06-21 13:31:43 -07:00 committed by GitHub
parent 6e29e185f3
commit 52e2bdf541
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
15 changed files with 40 additions and 13 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

1
onnxruntime/contrib_ops/cuda/bert/add_bias_transpose.cu
View file

@ -1,5 +1,6 @@
// Copyright (c) Microsoft Corporation. All rights reserved.
// Licensed under the MIT License.
#include "core/providers/cuda/cuda_common.h"
#include "core/providers/cuda/cu_inc/common.cuh"
#include "contrib_ops/cuda/bert/add_bias_transpose.h"

1
onnxruntime/contrib_ops/cuda/bert/embed_layer_norm_impl.h
View file

@ -1,5 +1,6 @@
// Copyright (c) Microsoft Corporation. All rights reserved.
// Licensed under the MIT License.
#pragma once
#include "core/common/common.h"

1
onnxruntime/contrib_ops/cuda/bert/transformer_common.cc
View file

@ -1,5 +1,6 @@
// Copyright (c) Microsoft Corporation. All rights reserved.
// Licensed under the MIT License.
#include <iostream>
#include "core/providers/shared_library/provider_api.h" // Include this otherwise Windows build complains Env::Default() missing
#include "core/platform/env_var_utils.h"

1
onnxruntime/contrib_ops/cuda/collective/nccl_kernels.cc
View file

@ -1,5 +1,6 @@
// Copyright (c) Microsoft Corporation. All rights reserved.
// Licensed under the MIT License.
#include "nccl_kernels.h"
#include "mpi_include.h"
#include "core/providers/cuda/tensor/transpose.h"

3
onnxruntime/contrib_ops/cuda/math/binary_elementwise_ops_impl.h
View file

@ -1,3 +1,6 @@
// Copyright (c) Microsoft Corporation. All rights reserved.
// Licensed under the MIT License.
#include <stdint.h>
#include "core/providers/cuda/shared_inc/cuda_utils.h"

3
onnxruntime/contrib_ops/cuda/quantization/qordered_ops/qordered_qdq.cc
View file

@ -1,4 +1,5 @@
// Copyright (c) Microsoft Corporation. All rights reserved.
// Licensed under the MIT License.
#include "core/framework/tensor_shape.h"
#include "contrib_ops/cuda/quantization/qordered_ops/qordered_qdq.h"

3
onnxruntime/core/providers/cuda/cu_inc/variadic_elementwise_impl.cuh
View file

@ -1,3 +1,6 @@
// Copyright (c) Microsoft Corporation. All rights reserved.
// Licensed under the MIT License.
#pragma once
#include "core/providers/cuda/cu_inc/common.cuh"

3
onnxruntime/core/providers/cuda/cuda_provider_interface.cc
View file

@ -1,3 +1,6 @@
// Copyright (c) Microsoft Corporation. All rights reserved.
// Licensed under the MIT License.
#include "core/session/onnxruntime_c_api.h"
namespace onnxruntime {

3
onnxruntime/core/providers/cuda/cuda_stream_handle.cc
View file

@ -1,5 +1,6 @@
//// Copyright (c) Microsoft Corporation. All rights reserved.
// Copyright (c) Microsoft Corporation. All rights reserved.
// Licensed under the MIT License.
#include "core/providers/cuda/cuda_stream_handle.h"
#include "core/providers/cuda/cuda_common.h"
#include "core/common/spin_pause.h"

3
onnxruntime/core/providers/cuda/cuda_stream_handle.h
View file

@ -1,5 +1,6 @@
//// Copyright (c) Microsoft Corporation. All rights reserved.
// Copyright (c) Microsoft Corporation. All rights reserved.
// Licensed under the MIT License.
#pragma once
#include "core/providers/cuda/cuda_pch.h"
#include "core/providers/cuda/shared_inc/cuda_utils.h"

3
onnxruntime/core/providers/cuda/cupti_manager.cc
View file

@ -1,3 +1,6 @@
// Copyright (c) Microsoft Corporation. All rights reserved.
// Licensed under the MIT License.
#include "cupti_manager.h"
#include <memory>

3
onnxruntime/core/providers/cuda/cupti_manager.h
View file

@ -1,3 +1,6 @@
// Copyright (c) Microsoft Corporation. All rights reserved.
// Licensed under the MIT License.
#pragma once
#if defined(USE_CUDA) && defined(ENABLE_CUDA_PROFILING)

1
onnxruntime/core/providers/cuda/generator/constant_of_shape.cc
View file

@ -1,5 +1,6 @@
// Copyright (c) Microsoft Corporation. All rights reserved.
// Licensed under the MIT License.
#include "constant_of_shape.h"
using namespace ::onnxruntime::common;

23
onnxruntime/core/providers/cuda/tensor/resize_impl.cu
View file

@ -1,3 +1,6 @@
// Copyright (c) Microsoft Corporation. All rights reserved.
// Licensed under the MIT License.
#include "core/providers/cuda/cu_inc/common.cuh"
#include "core/providers/cuda/tensor/resize_impl.h"
@ -156,7 +159,7 @@ __global__ void _ResizeNearestMappingKernel2D(
extrapolation_enabled && (orig_coord < 0.f || orig_coord > static_cast<float>(input_height - 1)));
dim = calc_nearest_pixel(orig_coord, scales_height < 1);
if (dim >= input_height) dim = input_height - 1;
if (dim < 0) dim = 0;
if (dim < 0) dim = 0;
}
dims_mapping[id].origin_ = dim;
@ -173,7 +176,7 @@ __global__ void _ResizeNearestMappingKernel2D(
extrapolation_enabled && (orig_coord < 0.f || orig_coord > static_cast<float>(input_width - 1)));
dim = calc_nearest_pixel(orig_coord, scales_width < 1);
if (dim >= input_width) dim = input_width - 1;
if (dim < 0) dim = 0;
if (dim < 0) dim = 0;
}
dims_mapping[id].origin_ = dim;
@ -212,7 +215,7 @@ __global__ void _ResizeNearestMappingKernel(
dims_mapping[id].extrapolate_ = static_cast<int>(extrapolation_enabled && (orig_coord < 0.f || orig_coord > static_cast<float>(input_shape[axis] - 1)));
dim = calc_nearest_pixel(orig_coord, scales[axis] < 1);
if (dim >= input_shape[axis]) dim = input_shape[axis] - 1;
if (dim < 0) dim = 0;
if (dim < 0) dim = 0;
}
dims_mapping[id].origin_ = dim;
@ -378,9 +381,9 @@ __global__ void _ResizeTrilinearCoordinateMapping(
dims_mapping[id].origin_ = z_int;
dims_mapping[id].weight_ = (z_int >= input_depth - 1) ? 0.5f : input_z - z_int;
} else if (id >= output_depth && id < (output_depth + output_height)) { // y = id - output_depth
float input_y = scale_height == 1 ? static_cast<float>(id - output_depth) :
transform_coordinate(static_cast<float>(id - output_depth), scale_height,
static_cast<float>(output_height), static_cast<float>(input_height),
float input_y = scale_height == 1 ? static_cast<float>(id - output_depth) :
transform_coordinate(static_cast<float>(id - output_depth), scale_height,
static_cast<float>(output_height), static_cast<float>(input_height),
roi_height_start, roi_height_end);
dims_mapping[id].extrapolate_ = (int)(extrapolation_enabled && (input_y < 0 || input_y > static_cast<float>(input_height - 1)));
@ -416,12 +419,12 @@ __global__ void _ResizeTrilinearKernel(
div_output_image.divmod(id, bxc, output_image_index);
CUDA_LONG input_index = bxc * input_depth * input_height * input_width;
int output_z, output_y, output_x, temp;
div_output_height.divmod(output_image_index, output_z, temp);
div_output_width.divmod(temp, output_y, output_x);
if (dims_mapping[output_z].extrapolate_ ||
dims_mapping[output_y + output_depth].extrapolate_ ||
if (dims_mapping[output_z].extrapolate_ ||
dims_mapping[output_y + output_depth].extrapolate_ ||
dims_mapping[output_x + output_depth + output_height].extrapolate_) {
output_data[id] = extrapolation_value;
return;
@ -435,7 +438,7 @@ __global__ void _ResizeTrilinearKernel(
float x_offset_0 = dims_mapping[output_x + output_depth + output_height].weight_;
int x_int = dims_mapping[output_x + output_depth + output_height].origin_;
input_index += z_int * input_height * input_width + y_int * input_width + x_int;
T x000 = input_data[input_index];

1
onnxruntime/core/providers/cuda/tensor/scatter_elements.h
View file

@ -1,5 +1,6 @@
// Copyright (c) Microsoft Corporation. All rights reserved.
// Licensed under the MIT License.
#pragma once
#include "core/providers/shared_library/provider_api.h"