I am updating the BatchNormalizationUpdateOutputInference_kernel by adding some custom arguments for the new version.
The set of files that are updated are as follows -
- ./aten/src/THNN/generic/BatchNormalization.c
- ./aten/src/THCUNN/BatchNormalization.cu
- ./aten/src/THCUNN/generic/BatchNormalization.cu
- ./torch/legacy/nn/BatchNormalization.py
- ./aten/src/THNN/generic/THNN.h
- ./aten/src/THCUNN/generic/THCUNN.h
But stills the python setup.py build fails by
In file, ./pytorch/aten/src/ATen/nn_parse.py
RuntimeError: BatchNormalization_updateOutput: can’t find binding