[Docs] Add header for files (#1181)

* [Docs] Add header for files

* change to OpenMMLab

* add headers for .cpp, .cu, .h, .cuh

* replace Open-MMLab with OpenMMLab
pull/1206/head^2
Jerry Jiarui XU 4 years ago committed by GitHub
parent 04daea425b
commit c47c9196d0
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
  1. 2
      LICENSE
  2. 2
      mmcv/__init__.py
  3. 2
      mmcv/arraymisc/__init__.py
  4. 2
      mmcv/arraymisc/quantization.py
  5. 2
      mmcv/cnn/__init__.py
  6. 2
      mmcv/cnn/alexnet.py
  7. 1
      mmcv/cnn/bricks/__init__.py
  8. 1
      mmcv/cnn/bricks/activation.py
  9. 1
      mmcv/cnn/bricks/context_block.py
  10. 1
      mmcv/cnn/bricks/conv.py
  11. 1
      mmcv/cnn/bricks/conv2d_adaptive_padding.py
  12. 1
      mmcv/cnn/bricks/conv_module.py
  13. 1
      mmcv/cnn/bricks/conv_ws.py
  14. 1
      mmcv/cnn/bricks/depthwise_separable_conv_module.py
  15. 1
      mmcv/cnn/bricks/drop.py
  16. 1
      mmcv/cnn/bricks/generalized_attention.py
  17. 1
      mmcv/cnn/bricks/hsigmoid.py
  18. 1
      mmcv/cnn/bricks/hswish.py
  19. 1
      mmcv/cnn/bricks/non_local.py
  20. 1
      mmcv/cnn/bricks/norm.py
  21. 1
      mmcv/cnn/bricks/padding.py
  22. 1
      mmcv/cnn/bricks/registry.py
  23. 1
      mmcv/cnn/bricks/scale.py
  24. 1
      mmcv/cnn/bricks/swish.py
  25. 1
      mmcv/cnn/bricks/transformer.py
  26. 1
      mmcv/cnn/bricks/upsample.py
  27. 1
      mmcv/cnn/bricks/wrappers.py
  28. 1
      mmcv/cnn/builder.py
  29. 2
      mmcv/cnn/resnet.py
  30. 2
      mmcv/cnn/utils/__init__.py
  31. 1
      mmcv/cnn/utils/fuse_conv_bn.py
  32. 2
      mmcv/cnn/utils/weight_init.py
  33. 2
      mmcv/cnn/vgg.py
  34. 1
      mmcv/engine/__init__.py
  35. 1
      mmcv/engine/test.py
  36. 2
      mmcv/fileio/__init__.py
  37. 1
      mmcv/fileio/file_client.py
  38. 2
      mmcv/fileio/handlers/__init__.py
  39. 2
      mmcv/fileio/handlers/base.py
  40. 2
      mmcv/fileio/handlers/json_handler.py
  41. 2
      mmcv/fileio/handlers/pickle_handler.py
  42. 2
      mmcv/fileio/handlers/yaml_handler.py
  43. 2
      mmcv/fileio/io.py
  44. 2
      mmcv/fileio/parse.py
  45. 2
      mmcv/image/__init__.py
  46. 2
      mmcv/image/colorspace.py
  47. 2
      mmcv/image/geometric.py
  48. 2
      mmcv/image/io.py
  49. 1
      mmcv/image/misc.py
  50. 1
      mmcv/image/photometric.py
  51. 1
      mmcv/onnx/__init__.py
  52. 1
      mmcv/onnx/info.py
  53. 1
      mmcv/onnx/onnx_utils/__init__.py
  54. 1
      mmcv/onnx/onnx_utils/symbolic_helper.py
  55. 1
      mmcv/onnx/symbolic.py
  56. 1
      mmcv/ops/__init__.py
  57. 1
      mmcv/ops/bbox.py
  58. 1
      mmcv/ops/border_align.py
  59. 1
      mmcv/ops/box_iou_rotated.py
  60. 1
      mmcv/ops/carafe.py
  61. 1
      mmcv/ops/cc_attention.py
  62. 1
      mmcv/ops/contour_expand.py
  63. 1
      mmcv/ops/corner_pool.py
  64. 1
      mmcv/ops/csrc/bbox_overlaps_cuda_kernel.cuh
  65. 1
      mmcv/ops/csrc/border_align_cuda_kernel.cuh
  66. 1
      mmcv/ops/csrc/carafe_cuda_kernel.cuh
  67. 1
      mmcv/ops/csrc/carafe_naive_cuda_kernel.cuh
  68. 1
      mmcv/ops/csrc/cc_attention_cuda_kernel.cuh
  69. 1
      mmcv/ops/csrc/deform_roi_pool_cuda_kernel.cuh
  70. 1
      mmcv/ops/csrc/masked_conv2d_cuda_kernel.cuh
  71. 1
      mmcv/ops/csrc/nms_cuda_kernel.cuh
  72. 1
      mmcv/ops/csrc/onnxruntime/corner_pool.h
  73. 1
      mmcv/ops/csrc/onnxruntime/cpu/corner_pool.cpp
  74. 1
      mmcv/ops/csrc/onnxruntime/cpu/gridSample.cpp
  75. 1
      mmcv/ops/csrc/onnxruntime/cpu/nms.cpp
  76. 1
      mmcv/ops/csrc/onnxruntime/cpu/onnxruntime_register.cpp
  77. 1
      mmcv/ops/csrc/onnxruntime/cpu/reduce_ops.cpp
  78. 1
      mmcv/ops/csrc/onnxruntime/cpu/roi_align.cpp
  79. 1
      mmcv/ops/csrc/onnxruntime/cpu/soft_nms.cpp
  80. 1
      mmcv/ops/csrc/onnxruntime/grid_sample.h
  81. 1
      mmcv/ops/csrc/onnxruntime/nms.h
  82. 1
      mmcv/ops/csrc/onnxruntime/onnxruntime_register.h
  83. 1
      mmcv/ops/csrc/onnxruntime/ort_mmcv_utils.h
  84. 1
      mmcv/ops/csrc/onnxruntime/reduce_ops.h
  85. 1
      mmcv/ops/csrc/onnxruntime/roi_align.h
  86. 1
      mmcv/ops/csrc/onnxruntime/roi_align_rotated.h
  87. 1
      mmcv/ops/csrc/onnxruntime/soft_nms.h
  88. 1
      mmcv/ops/csrc/parrots/bbox_overlaps.cpp
  89. 1
      mmcv/ops/csrc/parrots/bbox_overlaps_cuda.cu
  90. 1
      mmcv/ops/csrc/parrots/bbox_overlaps_parrots.cpp
  91. 1
      mmcv/ops/csrc/parrots/bbox_overlaps_pytorch.h
  92. 1
      mmcv/ops/csrc/parrots/border_align.cpp
  93. 1
      mmcv/ops/csrc/parrots/border_align_cuda.cu
  94. 1
      mmcv/ops/csrc/parrots/border_align_parrots.cpp
  95. 1
      mmcv/ops/csrc/parrots/border_align_pytorch.h
  96. 1
      mmcv/ops/csrc/parrots/box_iou_rotated_parrots.cpp
  97. 1
      mmcv/ops/csrc/parrots/box_iou_rotated_pytorch.h
  98. 1
      mmcv/ops/csrc/parrots/carafe.cpp
  99. 1
      mmcv/ops/csrc/parrots/carafe_cuda.cu
  100. 1
      mmcv/ops/csrc/parrots/carafe_naive.cpp
  101. Some files were not shown because too many files have changed in this diff Show More

@ -1,4 +1,4 @@
Copyright (c) Open-MMLab. All rights reserved.
Copyright (c) OpenMMLab. All rights reserved
Apache License
Version 2.0, January 2004

@ -1,4 +1,4 @@
# Copyright (c) Open-MMLab. All rights reserved.
# Copyright (c) OpenMMLab. All rights reserved.
# flake8: noqa
from .arraymisc import *
from .fileio import *

@ -1,4 +1,4 @@
# Copyright (c) Open-MMLab. All rights reserved.
# Copyright (c) OpenMMLab. All rights reserved.
from .quantization import dequantize, quantize
__all__ = ['quantize', 'dequantize']

@ -1,4 +1,4 @@
# Copyright (c) Open-MMLab. All rights reserved.
# Copyright (c) OpenMMLab. All rights reserved.
import numpy as np

@ -1,4 +1,4 @@
# Copyright (c) Open-MMLab. All rights reserved.
# Copyright (c) OpenMMLab. All rights reserved.
from .alexnet import AlexNet
# yapf: disable
from .bricks import (ACTIVATION_LAYERS, CONV_LAYERS, NORM_LAYERS,

@ -1,4 +1,4 @@
# Copyright (c) Open-MMLab. All rights reserved.
# Copyright (c) OpenMMLab. All rights reserved.
import logging
import torch.nn as nn

@ -1,3 +1,4 @@
# Copyright (c) OpenMMLab. All rights reserved.
from .activation import build_activation_layer
from .context_block import ContextBlock
from .conv import build_conv_layer

@ -1,3 +1,4 @@
# Copyright (c) OpenMMLab. All rights reserved.
import torch
import torch.nn as nn
import torch.nn.functional as F

@ -1,3 +1,4 @@
# Copyright (c) OpenMMLab. All rights reserved.
import torch
from torch import nn

@ -1,3 +1,4 @@
# Copyright (c) OpenMMLab. All rights reserved.
from torch import nn
from .registry import CONV_LAYERS

@ -1,3 +1,4 @@
# Copyright (c) OpenMMLab. All rights reserved.
import math
from torch import nn

@ -1,3 +1,4 @@
# Copyright (c) OpenMMLab. All rights reserved.
import warnings
import torch.nn as nn

@ -1,3 +1,4 @@
# Copyright (c) OpenMMLab. All rights reserved.
import torch
import torch.nn as nn
import torch.nn.functional as F

@ -1,3 +1,4 @@
# Copyright (c) OpenMMLab. All rights reserved.
import torch.nn as nn
from .conv_module import ConvModule

@ -1,3 +1,4 @@
# Copyright (c) OpenMMLab. All rights reserved.
import torch
import torch.nn as nn

@ -1,3 +1,4 @@
# Copyright (c) OpenMMLab. All rights reserved.
import math
import numpy as np

@ -1,3 +1,4 @@
# Copyright (c) OpenMMLab. All rights reserved.
import torch.nn as nn
from .registry import ACTIVATION_LAYERS

@ -1,3 +1,4 @@
# Copyright (c) OpenMMLab. All rights reserved.
import torch.nn as nn
from .registry import ACTIVATION_LAYERS

@ -1,3 +1,4 @@
# Copyright (c) OpenMMLab. All rights reserved.
from abc import ABCMeta
import torch

@ -1,3 +1,4 @@
# Copyright (c) OpenMMLab. All rights reserved.
import inspect
import torch.nn as nn

@ -1,3 +1,4 @@
# Copyright (c) OpenMMLab. All rights reserved.
import torch.nn as nn
from .registry import PADDING_LAYERS

@ -1,3 +1,4 @@
# Copyright (c) OpenMMLab. All rights reserved.
from mmcv.utils import Registry
CONV_LAYERS = Registry('conv layer')

@ -1,3 +1,4 @@
# Copyright (c) OpenMMLab. All rights reserved.
import torch
import torch.nn as nn

@ -1,3 +1,4 @@
# Copyright (c) OpenMMLab. All rights reserved.
import torch
import torch.nn as nn

@ -1,3 +1,4 @@
# Copyright (c) OpenMMLab. All rights reserved.
import copy
import warnings

@ -1,3 +1,4 @@
# Copyright (c) OpenMMLab. All rights reserved.
import torch.nn as nn
import torch.nn.functional as F

@ -1,3 +1,4 @@
# Copyright (c) OpenMMLab. All rights reserved.
r"""Modified from https://github.com/facebookresearch/detectron2/blob/master/detectron2/layers/wrappers.py # noqa: E501
Wrap some nn modules to support empty tensor input. Currently, these wrappers

@ -1,3 +1,4 @@
# Copyright (c) OpenMMLab. All rights reserved.
from ..runner import Sequential
from ..utils import Registry, build_from_cfg

@ -1,4 +1,4 @@
# Copyright (c) Open-MMLab. All rights reserved.
# Copyright (c) OpenMMLab. All rights reserved.
import logging
import torch.nn as nn

@ -1,4 +1,4 @@
# Copyright (c) Open-MMLab. All rights reserved.
# Copyright (c) OpenMMLab. All rights reserved.
from .flops_counter import get_model_complexity_info
from .fuse_conv_bn import fuse_conv_bn
from .weight_init import (INITIALIZERS, Caffe2XavierInit, ConstantInit,

@ -1,3 +1,4 @@
# Copyright (c) OpenMMLab. All rights reserved.
import torch
import torch.nn as nn

@ -1,4 +1,4 @@
# Copyright (c) Open-MMLab. All rights reserved.
# Copyright (c) OpenMMLab. All rights reserved.
import copy
import math
import warnings

@ -1,4 +1,4 @@
# Copyright (c) Open-MMLab. All rights reserved.
# Copyright (c) OpenMMLab. All rights reserved.
import logging
import torch.nn as nn

@ -1,3 +1,4 @@
# Copyright (c) OpenMMLab. All rights reserved.
from .test import (collect_results_cpu, collect_results_gpu, multi_gpu_test,
single_gpu_test)

@ -1,3 +1,4 @@
# Copyright (c) OpenMMLab. All rights reserved.
import os.path as osp
import pickle
import shutil

@ -1,4 +1,4 @@
# Copyright (c) Open-MMLab. All rights reserved.
# Copyright (c) OpenMMLab. All rights reserved.
from .file_client import BaseStorageBackend, FileClient
from .handlers import BaseFileHandler, JsonHandler, PickleHandler, YamlHandler
from .io import dump, load, register_handler

@ -1,3 +1,4 @@
# Copyright (c) OpenMMLab. All rights reserved.
import inspect
from abc import ABCMeta, abstractmethod
from urllib.request import urlopen

@ -1,4 +1,4 @@
# Copyright (c) Open-MMLab. All rights reserved.
# Copyright (c) OpenMMLab. All rights reserved.
from .base import BaseFileHandler
from .json_handler import JsonHandler
from .pickle_handler import PickleHandler

@ -1,4 +1,4 @@
# Copyright (c) Open-MMLab. All rights reserved.
# Copyright (c) OpenMMLab. All rights reserved.
from abc import ABCMeta, abstractmethod

@ -1,4 +1,4 @@
# Copyright (c) Open-MMLab. All rights reserved.
# Copyright (c) OpenMMLab. All rights reserved.
import json
import numpy as np

@ -1,4 +1,4 @@
# Copyright (c) Open-MMLab. All rights reserved.
# Copyright (c) OpenMMLab. All rights reserved.
import pickle
from .base import BaseFileHandler

@ -1,4 +1,4 @@
# Copyright (c) Open-MMLab. All rights reserved.
# Copyright (c) OpenMMLab. All rights reserved.
import yaml
try:

@ -1,4 +1,4 @@
# Copyright (c) Open-MMLab. All rights reserved.
# Copyright (c) OpenMMLab. All rights reserved.
from pathlib import Path
from ..utils import is_list_of, is_str

@ -1,4 +1,4 @@
# Copyright (c) Open-MMLab. All rights reserved.
# Copyright (c) OpenMMLab. All rights reserved.
def list_from_file(filename, prefix='', offset=0, max_num=0, encoding='utf-8'):
"""Load a text file and parse the content as a list of strings.

@ -1,4 +1,4 @@
# Copyright (c) Open-MMLab. All rights reserved.
# Copyright (c) OpenMMLab. All rights reserved.
from .colorspace import (bgr2gray, bgr2hls, bgr2hsv, bgr2rgb, bgr2ycbcr,
gray2bgr, gray2rgb, hls2bgr, hsv2bgr, imconvert,
rgb2bgr, rgb2gray, rgb2ycbcr, ycbcr2bgr, ycbcr2rgb)

@ -1,4 +1,4 @@
# Copyright (c) Open-MMLab. All rights reserved.
# Copyright (c) OpenMMLab. All rights reserved.
import cv2
import numpy as np

@ -1,4 +1,4 @@
# Copyright (c) Open-MMLab. All rights reserved.
# Copyright (c) OpenMMLab. All rights reserved.
import numbers
import cv2

@ -1,4 +1,4 @@
# Copyright (c) Open-MMLab. All rights reserved.
# Copyright (c) OpenMMLab. All rights reserved.
import io
import os.path as osp
from pathlib import Path

@ -1,3 +1,4 @@
# Copyright (c) OpenMMLab. All rights reserved.
import numpy as np
import mmcv

@ -1,3 +1,4 @@
# Copyright (c) OpenMMLab. All rights reserved.
import cv2
import numpy as np

@ -1,3 +1,4 @@
# Copyright (c) OpenMMLab. All rights reserved.
from .info import is_custom_op_loaded
from .symbolic import register_extra_symbolics

@ -1,3 +1,4 @@
# Copyright (c) OpenMMLab. All rights reserved.
import os

@ -0,0 +1 @@
# Copyright (c) OpenMMLab. All rights reserved.

@ -1,3 +1,4 @@
# Copyright (c) OpenMMLab. All rights reserved.
"""Modified from https://github.com/pytorch/pytorch."""
import warnings
from functools import wraps

@ -1,3 +1,4 @@
# Copyright (c) OpenMMLab. All rights reserved.
"""Modified from https://github.com/pytorch/pytorch."""
import os

@ -1,3 +1,4 @@
# Copyright (c) OpenMMLab. All rights reserved.
from .bbox import bbox_overlaps
from .border_align import BorderAlign, border_align
from .box_iou_rotated import box_iou_rotated

@ -1,3 +1,4 @@
# Copyright (c) OpenMMLab. All rights reserved.
from ..utils import ext_loader
ext_module = ext_loader.load_ext('_ext', ['bbox_overlaps'])

@ -1,3 +1,4 @@
# Copyright (c) OpenMMLab. All rights reserved.
# modified from
# https://github.com/Megvii-BaseDetection/cvpods/blob/master/cvpods/layers/border_align.py

@ -1,3 +1,4 @@
# Copyright (c) OpenMMLab. All rights reserved.
from ..utils import ext_loader
ext_module = ext_loader.load_ext('_ext', ['box_iou_rotated'])

@ -1,3 +1,4 @@
# Copyright (c) OpenMMLab. All rights reserved.
import torch
import torch.nn as nn
import torch.nn.functional as F

@ -1,3 +1,4 @@
# Copyright (c) OpenMMLab. All rights reserved.
import torch
import torch.nn as nn
import torch.nn.functional as F

@ -1,3 +1,4 @@
# Copyright (c) OpenMMLab. All rights reserved.
import numpy as np
import torch

@ -1,3 +1,4 @@
# Copyright (c) OpenMMLab. All rights reserved.
import torch
from torch import nn
from torch.autograd import Function

@ -1,3 +1,4 @@
// Copyright (c) OpenMMLab. All rights reserved
#ifndef BBOX_OVERLAPS_CUDA_KERNEL_CUH
#define BBOX_OVERLAPS_CUDA_KERNEL_CUH

@ -1,3 +1,4 @@
// Copyright (c) OpenMMLab. All rights reserved
// modified from
// https://github.com/Megvii-BaseDetection/cvpods/blob/master/cvpods/layers/csrc/border_align/border_align_kernel.cu.
// the main difference: (1) use `argmax_idx` for fast computing of gradient

@ -1,3 +1,4 @@
// Copyright (c) OpenMMLab. All rights reserved
#ifndef CARAFE_CUDA_KERNEL_CUH
#define CARAFE_CUDA_KERNEL_CUH

@ -1,3 +1,4 @@
// Copyright (c) OpenMMLab. All rights reserved
#ifndef CARAFE_NAIVE_CUDA_KERNEL_CUH
#define CARAFE_NAIVE_CUDA_KERNEL_CUH

@ -1,3 +1,4 @@
// Copyright (c) OpenMMLab. All rights reserved
#ifndef CC_ATTENTION_CUDA_KERNEL_CUH
#define CC_ATTENTION_CUDA_KERNEL_CUH

@ -1,3 +1,4 @@
// Copyright (c) OpenMMLab. All rights reserved
#ifndef DEFORM_ROI_POOL_CUDA_KERNEL_CUH
#define DEFORM_ROI_POOL_CUDA_KERNEL_CUH

@ -1,3 +1,4 @@
// Copyright (c) OpenMMLab. All rights reserved
#ifndef MASKED_CONV2D_CUDA_KERNEL_CUH
#define MASKED_CONV2D_CUDA_KERNEL_CUH

@ -1,3 +1,4 @@
// Copyright (c) OpenMMLab. All rights reserved
#ifndef NMS_CUDA_KERNEL_CUH
#define NMS_CUDA_KERNEL_CUH

@ -1,3 +1,4 @@
// Copyright (c) OpenMMLab. All rights reserved
#ifndef ONNXRUNTIME_CORNER_POOL_H
#define ONNXRUNTIME_CORNER_POOL_H

@ -1,3 +1,4 @@
// Copyright (c) OpenMMLab. All rights reserved
#include "corner_pool.h"
#include "../ort_mmcv_utils.h"

@ -1,3 +1,4 @@
// Copyright (c) OpenMMLab. All rights reserved
#include <cmath>
#include "../ort_mmcv_utils.h"

@ -1,3 +1,4 @@
// Copyright (c) OpenMMLab. All rights reserved
#include "nms.h"
#include <assert.h>

@ -1,3 +1,4 @@
// Copyright (c) OpenMMLab. All rights reserved
#include "onnxruntime_register.h"
#include "corner_pool.h"

@ -1,3 +1,4 @@
// Copyright (c) OpenMMLab. All rights reserved
#include "reduce_ops.h"
#include <assert.h>

@ -1,3 +1,4 @@
// Copyright (c) OpenMMLab. All rights reserved
#include "roi_align.h"
#include "../ort_mmcv_utils.h"

@ -1,3 +1,4 @@
// Copyright (c) OpenMMLab. All rights reserved
#include "soft_nms.h"
#include <assert.h>

@ -1,3 +1,4 @@
// Copyright (c) OpenMMLab. All rights reserved
#ifndef ONNXRUNTIME_GRIDSAMPLE_H
#define ONNXRUNTIME_GRIDSAMPLE_H

@ -1,3 +1,4 @@
// Copyright (c) OpenMMLab. All rights reserved
#ifndef ONNXRUNTIME_NMS_H
#define ONNXRUNTIME_NMS_H

@ -1,3 +1,4 @@
// Copyright (c) OpenMMLab. All rights reserved
#ifndef ONNXRUNTIME_REGISTER_H
#define ONNXRUNTIME_REGISTER_H
#include <onnxruntime_c_api.h>

@ -1,3 +1,4 @@
// Copyright (c) OpenMMLab. All rights reserved
#ifndef ORT_MMCV_UTILS_H
#define ORT_MMCV_UTILS_H
#include <onnxruntime_cxx_api.h>

@ -1,3 +1,4 @@
// Copyright (c) OpenMMLab. All rights reserved
#ifndef ONNXRUNTIME_REDUCE_OPS_H
#define ONNXRUNTIME_REDUCE_OPS_H

@ -1,3 +1,4 @@
// Copyright (c) OpenMMLab. All rights reserved
#ifndef ONNXRUNTIME_ROI_ALIGN_H
#define ONNXRUNTIME_ROI_ALIGN_H

@ -1,3 +1,4 @@
// Copyright (c) OpenMMLab. All rights reserved
#ifndef ONNXRUNTIME_ROI_ALIGN_ROTATED_H
#define ONNXRUNTIME_ROI_ALIGN_ROTATED_H

@ -1,3 +1,4 @@
// Copyright (c) OpenMMLab. All rights reserved
#ifndef ONNXRUNTIME_SOFT_NMS_H
#define ONNXRUNTIME_SOFT_NMS_H
#include <onnxruntime_cxx_api.h>

@ -1,3 +1,4 @@
// Copyright (c) OpenMMLab. All rights reserved
#include "pytorch_cpp_helper.hpp"
#ifdef MMCV_WITH_CUDA

@ -1,3 +1,4 @@
// Copyright (c) OpenMMLab. All rights reserved
#include "bbox_overlaps_cuda_kernel.cuh"
#include "pytorch_cuda_helper.hpp"

@ -1,3 +1,4 @@
// Copyright (c) OpenMMLab. All rights reserved
#include <parrots/compute/aten.hpp>
#include <parrots/extension.hpp>
#include <parrots/foundation/ssattrs.hpp>

@ -1,3 +1,4 @@
// Copyright (c) OpenMMLab. All rights reserved
#ifndef BBOX_OVERLAPS_PYTORCH_H
#define BBOX_OVERLAPS_PYTORCH_H
#include <torch/extension.h>

@ -1,3 +1,4 @@
// Copyright (c) OpenMMLab. All rights reserved
#include "pytorch_cpp_helper.hpp"
#ifdef MMCV_WITH_CUDA

@ -1,3 +1,4 @@
// Copyright (c) OpenMMLab. All rights reserved
#include "border_align_cuda_kernel.cuh"
#include "pytorch_cuda_helper.hpp"

@ -1,3 +1,4 @@
// Copyright (c) OpenMMLab. All rights reserved
#include <parrots/compute/aten.hpp>
#include <parrots/extension.hpp>
#include <parrots/foundation/ssattrs.hpp>

@ -1,3 +1,4 @@
// Copyright (c) OpenMMLab. All rights reserved
#ifndef BORDER_ALIGN_PYTORCH_H
#define BORDER_ALIGN_PYTORCH_H
#include <torch/extension.h>

@ -1,3 +1,4 @@
// Copyright (c) OpenMMLab. All rights reserved
#include <parrots/compute/aten.hpp>
#include <parrots/extension.hpp>
#include <parrots/foundation/ssattrs.hpp>

@ -1,3 +1,4 @@
// Copyright (c) OpenMMLab. All rights reserved
#ifndef BOX_IOU_ROTATED_PYTORCH_H
#define BOX_IOU_ROTATED_PYTORCH_H
#include <torch/extension.h>

@ -1,3 +1,4 @@
// Copyright (c) OpenMMLab. All rights reserved
#include "pytorch_cpp_helper.hpp"
#ifdef MMCV_WITH_CUDA

@ -1,3 +1,4 @@
// Copyright (c) OpenMMLab. All rights reserved
#include "carafe_cuda_kernel.cuh"
#include "pytorch_cuda_helper.hpp"

@ -1,3 +1,4 @@
// Copyright (c) OpenMMLab. All rights reserved
#include "pytorch_cpp_helper.hpp"
#ifdef MMCV_WITH_CUDA

Some files were not shown because too many files have changed in this diff Show More

Loading…
Cancel
Save