Removed / breaking in @25.09, gone in @26.09
RankInfoFormatterClassapex/__init__.py
RankInfoFormatter.formatMethodapex/__init__.py
FMHAClassapex/contrib/fmha/fmha.py
FMHA.__init__Methodapex/contrib/fmha/fmha.py
FMHA.forwardMethodapex/contrib/fmha/fmha.py
FMHAFunClassapex/contrib/fmha/fmha.py
FMHAFun.backwardMethodapex/contrib/fmha/fmha.py
FMHAFun.forwardMethodapex/contrib/fmha/fmha.py
GDSFileFunctionapex/contrib/gpu_direct_storage/__init__.py
EncdecMultiheadAttnClassapex/contrib/multihead_attn/encdec_multihead_attn.py
EncdecMultiheadAttn.__init__Methodapex/contrib/multihead_attn/encdec_multihead_attn.py
EncdecMultiheadAttn.forwardMethodapex/contrib/multihead_attn/encdec_multihead_attn.py
EncdecMultiheadAttn.reset_parametersMethodapex/contrib/multihead_attn/encdec_multihead_attn.py
jit_dropout_addFunctionapex/contrib/multihead_attn/encdec_multihead_attn.py
EncdecAttnFuncClassapex/contrib/multihead_attn/encdec_multihead_attn_func.py
EncdecAttnFunc.backwardMethodapex/contrib/multihead_attn/encdec_multihead_attn_func.py
EncdecAttnFunc.forwardMethodapex/contrib/multihead_attn/encdec_multihead_attn_func.py
FastEncdecAttnFuncClassex/contrib/multihead_attn/fast_encdec_multihead_attn_func.py
FastEncdecAttnFunc.backwardMethodex/contrib/multihead_attn/fast_encdec_multihead_attn_func.py
FastEncdecAttnFunc.forwardMethodex/contrib/multihead_attn/fast_encdec_multihead_attn_func.py
FastEncdecAttnNormAddFuncClassb/multihead_attn/fast_encdec_multihead_attn_norm_add_func.py
FastEncdecAttnNormAddFunc.backwardMethodb/multihead_attn/fast_encdec_multihead_attn_norm_add_func.py
FastEncdecAttnNormAddFunc.forwardMethodb/multihead_attn/fast_encdec_multihead_attn_norm_add_func.py
FastSelfAttnFuncClassapex/contrib/multihead_attn/fast_self_multihead_attn_func.py
FastSelfAttnFunc.backwardMethodapex/contrib/multihead_attn/fast_self_multihead_attn_func.py
FastSelfAttnFunc.forwardMethodapex/contrib/multihead_attn/fast_self_multihead_attn_func.py
FastSelfAttnNormAddFuncClassrib/multihead_attn/fast_self_multihead_attn_norm_add_func.py
FastSelfAttnNormAddFunc.backwardMethodrib/multihead_attn/fast_self_multihead_attn_norm_add_func.py
FastSelfAttnNormAddFunc.forwardMethodrib/multihead_attn/fast_self_multihead_attn_norm_add_func.py
MaskSoftmaxDropoutClassapex/contrib/multihead_attn/mask_softmax_dropout_func.py
MaskSoftmaxDropout.backwardMethodapex/contrib/multihead_attn/mask_softmax_dropout_func.py
MaskSoftmaxDropout.forwardMethodapex/contrib/multihead_attn/mask_softmax_dropout_func.py
SelfMultiheadAttnClassapex/contrib/multihead_attn/self_multihead_attn.py
SelfMultiheadAttn.__init__Methodapex/contrib/multihead_attn/self_multihead_attn.py
SelfMultiheadAttn.forwardMethodapex/contrib/multihead_attn/self_multihead_attn.py
SelfMultiheadAttn.reset_parametersMethodapex/contrib/multihead_attn/self_multihead_attn.py
jit_dropout_addFunctionapex/contrib/multihead_attn/self_multihead_attn.py
SelfAttnFuncClassapex/contrib/multihead_attn/self_multihead_attn_func.py
SelfAttnFunc.backwardMethodapex/contrib/multihead_attn/self_multihead_attn_func.py
SelfAttnFunc.forwardMethodapex/contrib/multihead_attn/self_multihead_attn_func.py
TestFMHAClassapex/contrib/test/fmha/test_fmha.py
TestFMHA.run_testMethodapex/contrib/test/fmha/test_fmha.py
TestFMHA.test_128Methodapex/contrib/test/fmha/test_fmha.py
TestFMHA.test_256Methodapex/contrib/test/fmha/test_fmha.py
TestFMHA.test_384Methodapex/contrib/test/fmha/test_fmha.py
TestFMHA.test_512Methodapex/contrib/test/fmha/test_fmha.py
_get_device_propertiesFunctionapex/contrib/test/fmha/test_fmha.py
py_mhaFunctionapex/contrib/test/fmha/test_fmha.py
EncdecMultiheadAttnTestClassex/contrib/test/multihead_attn/test_encdec_multihead_attn.py
EncdecMultiheadAttnTest.setUpMethodex/contrib/test/multihead_attn/test_encdec_multihead_attn.py
EncdecMultiheadAttnTest.test_encdec_multihead_attnMethodex/contrib/test/multihead_attn/test_encdec_multihead_attn.py
EncdecMultiheadAttnTest.test_encdec_multihead_attn_pad_maskMethodex/contrib/test/multihead_attn/test_encdec_multihead_attn.py
EncdecMultiheadAttnTest.test_encdec_multihead_attn_time_maskMethodex/contrib/test/multihead_attn/test_encdec_multihead_attn.py
EncdecMultiheadAttnNormAddTestClassb/test/multihead_attn/test_encdec_multihead_attn_norm_add.py
EncdecMultiheadAttnNormAddTest.setUpMethodb/test/multihead_attn/test_encdec_multihead_attn_norm_add.py
EncdecMultiheadAttnNormAddTest.test_encdec_multihead_attn_norm_addMethodb/test/multihead_attn/test_encdec_multihead_attn_norm_add.py
SelfMultiheadAttnTestClassib/test/multihead_attn/test_fast_self_multihead_attn_bias.py
SelfMultiheadAttnTest.setUpMethodib/test/multihead_attn/test_fast_self_multihead_attn_bias.py
SelfMultiheadAttnTest.test_self_multihead_attn_additive_maskMethodib/test/multihead_attn/test_fast_self_multihead_attn_bias.py
FusedSoftmaxTestClassapex/contrib/test/multihead_attn/test_mha_fused_softmax.py
FusedSoftmaxTest.setUpMethodapex/contrib/test/multihead_attn/test_mha_fused_softmax.py
FusedSoftmaxTest.test_fused_softmaxMethodapex/contrib/test/multihead_attn/test_mha_fused_softmax.py
SelfMultiheadAttnTestClassapex/contrib/test/multihead_attn/test_self_multihead_attn.py
SelfMultiheadAttnTest.setUpMethodapex/contrib/test/multihead_attn/test_self_multihead_attn.py
SelfMultiheadAttnTest.test_self_multihead_attnMethodapex/contrib/test/multihead_attn/test_self_multihead_attn.py
SelfMultiheadAttnTest.test_self_multihead_attn_pad_maskMethodapex/contrib/test/multihead_attn/test_self_multihead_attn.py
SelfMultiheadAttnTest.test_self_multihead_attn_time_maskMethodapex/contrib/test/multihead_attn/test_self_multihead_attn.py
SelfMultiheadAttnNormAddTestClassrib/test/multihead_attn/test_self_multihead_attn_norm_add.py
SelfMultiheadAttnNormAddTest.setUpMethodrib/test/multihead_attn/test_self_multihead_attn_norm_add.py
SelfMultiheadAttnNormAddTest.test_self_multihead_attn_norm_addMethodrib/test/multihead_attn/test_self_multihead_attn_norm_add.py
MegatronPretrainingRandomSamplerClassapex/transformer/_data/_batchsampler.py
MegatronPretrainingRandomSampler.__init__Methodapex/transformer/_data/_batchsampler.py
MegatronPretrainingRandomSampler.__iter__Methodapex/transformer/_data/_batchsampler.py
MegatronPretrainingRandomSampler.__len__Methodapex/transformer/_data/_batchsampler.py
MegatronPretrainingRandomSampler.local_minibatch_sizeMethodapex/transformer/_data/_batchsampler.py
MegatronPretrainingSamplerClassapex/transformer/_data/_batchsampler.py
MegatronPretrainingSampler.__init__Methodapex/transformer/_data/_batchsampler.py
MegatronPretrainingSampler.__iter__Methodapex/transformer/_data/_batchsampler.py
MegatronPretrainingSampler.__len__Methodapex/transformer/_data/_batchsampler.py
MegatronPretrainingSampler.get_start_end_idxMethodapex/transformer/_data/_batchsampler.py
MegatronPretrainingSampler.local_minibatch_sizeMethodapex/transformer/_data/_batchsampler.py
_BaseClassapex/transformer/_data/_batchsampler.py
_Base.__iter__Methodapex/transformer/_data/_batchsampler.py
_Base.__len__Methodapex/transformer/_data/_batchsampler.py
_Base.local_minibatch_sizeMethodapex/transformer/_data/_batchsampler.py
GradScalerClassapex/transformer/amp/grad_scaler.py
GradScaler.__init__Methodapex/transformer/amp/grad_scaler.py
GradScaler._maybe_opt_stepMethodapex/transformer/amp/grad_scaler.py
GradScaler._unscale_grads_Methodapex/transformer/amp/grad_scaler.py
GradScaler.updateMethodapex/transformer/amp/grad_scaler.py
AttnMaskTypeClassapex/transformer/enums.py
AttnTypeClassapex/transformer/enums.py
LayerTypeClassapex/transformer/enums.py
ModelTypeClassapex/transformer/enums.py
FusedRoPE2DFuncClassapex/transformer/functional/fused_rope.py
FusedRoPE2DFunc.backwardMethodapex/transformer/functional/fused_rope.py
FusedRoPE2DFunc.forwardMethodapex/transformer/functional/fused_rope.py
FusedRoPECachedFuncClassapex/transformer/functional/fused_rope.py
FusedRoPECachedFunc.backwardMethodapex/transformer/functional/fused_rope.py
FusedRoPECachedFunc.forwardMethodapex/transformer/functional/fused_rope.py
… +670 more
Signature changed
_get_current_dtypeFunctionapex/_autocast_utils.py
Bottleneck.__init__Methodapex/contrib/bottleneck/bottleneck.py
SpatialBottleneck.__init__Methodapex/contrib/bottleneck/bottleneck.py
SpatialBottleneckFunction.forwardMethodapex/contrib/bottleneck/bottleneck.py
kaiming_uniform_Functionapex/contrib/bottleneck/bottleneck.py
HaloExchangerAllGather.left_right_halo_exchangeMethodapex/contrib/bottleneck/halo_exchangers.py
HaloExchangerNoComm.left_right_halo_exchangeMethodapex/contrib/bottleneck/halo_exchangers.py
HaloExchangerPeer.left_right_halo_exchangeMethodapex/contrib/bottleneck/halo_exchangers.py
HaloExchangerSendRecv.left_right_halo_exchangeMethodapex/contrib/bottleneck/halo_exchangers.py
clip_grad_norm_Functionapex/contrib/clip_grad/clip_grad.py
GroupBatchNorm2d.__init__Methodapex/contrib/cudnn_gbn/batch_norm.py
GroupBatchNorm2d.forwardMethodapex/contrib/cudnn_gbn/batch_norm.py
_GroupBatchNorm2d.forwardMethodapex/contrib/cudnn_gbn/batch_norm.py
GroupNorm.__init__Methodapex/contrib/group_norm/group_norm.py
fake_group_norm_nhwc_bpropFunctionapex/contrib/group_norm/group_norm.py
fake_group_norm_nhwc_fpropFunctionapex/contrib/group_norm/group_norm.py
group_norm_nhwc_bpropFunctionapex/contrib/group_norm/group_norm.py
group_norm_nhwc_fpropFunctionapex/contrib/group_norm/group_norm.py
BatchNorm2d_NHWC.__init__Methodapex/contrib/groupbn/batch_norm.py
bn_NHWC_impl.forwardMethodapex/contrib/groupbn/batch_norm.py
bn_addrelu_NHWC_impl.forwardMethodapex/contrib/groupbn/batch_norm.py
IndexMul2dBackward_.forwardMethodapex/contrib/index_mul_2d/index_mul_2d.py
nccl_mem.__init__Methodapex/contrib/nccl_allocator/nccl_allocator.py
_get_tuneable_triton_func_nameFunctionapex/contrib/openfold_triton/__init__.py
_load_triton_auto_tune_cacheFunctionapex/contrib/openfold_triton/__init__.py
FusedAdamSWA.__init__Methodapex/contrib/openfold_triton/fused_adam_swa.py
FusedAdamSWA.from_optimMethodapex/contrib/openfold_triton/fused_adam_swa.py
FusedAdamSWA.stepMethodapex/contrib/openfold_triton/fused_adam_swa.py
_attention_biasFunctionapex/contrib/openfold_triton/mha.py
DistributedFusedAdam.__init__Methodapex/contrib/optimizers/distributed_fused_adam.py
DistributedFusedAdam._check_params_shard_dtypesMethodapex/contrib/optimizers/distributed_fused_adam.py
DistributedFusedAdam._init_param_stateMethodapex/contrib/optimizers/distributed_fused_adam.py
DistributedFusedAdam._local_grad_normMethodapex/contrib/optimizers/distributed_fused_adam.py
DistributedFusedAdam._local_stepMethodapex/contrib/optimizers/distributed_fused_adam.py
DistributedFusedAdam._local_step_with_param_remaindersMethodapex/contrib/optimizers/distributed_fused_adam.py
DistributedFusedAdam._local_step_with_scaled_statesMethodapex/contrib/optimizers/distributed_fused_adam.py
DistributedFusedAdam._param_copyMethodapex/contrib/optimizers/distributed_fused_adam.py
DistributedFusedAdam._start_bucket_grad_syncMethodapex/contrib/optimizers/distributed_fused_adam.py
DistributedFusedAdam._start_bucket_param_syncMethodapex/contrib/optimizers/distributed_fused_adam.py
DistributedFusedAdam._try_start_bucket_grad_syncMethodapex/contrib/optimizers/distributed_fused_adam.py
DistributedFusedAdam.clip_grad_normMethodapex/contrib/optimizers/distributed_fused_adam.py
DistributedFusedAdam.grad_normMethodapex/contrib/optimizers/distributed_fused_adam.py
DistributedFusedAdam.init_paramsMethodapex/contrib/optimizers/distributed_fused_adam.py
DistributedFusedAdam.init_params_bucketMethodapex/contrib/optimizers/distributed_fused_adam.py
DistributedFusedAdam.parameterMethodapex/contrib/optimizers/distributed_fused_adam.py
DistributedFusedAdam.state_dictMethodapex/contrib/optimizers/distributed_fused_adam.py
DistributedFusedAdam.stepMethodapex/contrib/optimizers/distributed_fused_adam.py
DistributedFusedAdam.unscale_gradsMethodapex/contrib/optimizers/distributed_fused_adam.py
_multi_tensor_copyFunctionapex/contrib/optimizers/distributed_fused_adam.py
DistributedFusedLAMB.__init__Methodapex/contrib/optimizers/distributed_fused_lamb.py
FP16_Optimizer.__init__Methodapex/contrib/optimizers/fp16_optimizer.py
FusedAdam.__init__Methodapex/contrib/optimizers/fused_adam.py
FusedAdam.stepMethodapex/contrib/optimizers/fused_adam.py
FusedLAMB.__init__Methodapex/contrib/optimizers/fused_lamb.py
FusedSGD.__init__Methodapex/contrib/optimizers/fused_sgd.py
FusedSGD.stepMethodapex/contrib/optimizers/fused_sgd.py
ASP.init_model_for_pruningMethodapex/contrib/sparsity/asp.py
ASP.set_permutation_saving_paramsMethodapex/contrib/sparsity/asp.py
Permutation.build_fx_graphMethodapex/contrib/sparsity/permutation_lib.py
Permutation.fetch_attrMethodapex/contrib/sparsity/permutation_lib.py
Permutation.permute_modelMethodapex/contrib/sparsity/permutation_lib.py
Permutation.save_graph_to_jsonMethodapex/contrib/sparsity/permutation_lib.py
Permutation.set_permutation_saving_paramsMethodapex/contrib/sparsity/permutation_lib.py
use_swap_mapFunction/contrib/sparsity/permutation_search_kernels/channel_swap.py
build_stripe_mapFunctionrib/sparsity/permutation_search_kernels/exhaustive_search.py
generate_all_unique_combinationsFunctionrib/sparsity/permutation_search_kernels/exhaustive_search.py
generate_unique_combinationsFunctionrib/sparsity/permutation_search_kernels/exhaustive_search.py
use_gpuFunctionsparsity/permutation_search_kernels/permutation_utilities.py
compute_valid_1d_patternsFunctionapex/contrib/sparsity/sparse_masklib.py
compute_valid_2d_patternsFunctionapex/contrib/sparsity/sparse_masklib.py
MHA_test.__init__Methodapex/contrib/sparsity/test/test_permutation_application.py
already_sparse.__init__Methodapex/contrib/sparsity/test/test_permutation_application.py
conv_1d.__init__Methodapex/contrib/sparsity/test/test_permutation_application.py
simple_convs.__init__Methodapex/contrib/sparsity/test/test_permutation_application.py
test_concat.__init__Methodapex/contrib/sparsity/test/test_permutation_application.py
test_flatten_module.__init__Methodapex/contrib/sparsity/test/test_permutation_application.py
test_flatten_op.__init__Methodapex/contrib/sparsity/test/test_permutation_application.py
test_trace_failure.__init__Methodapex/contrib/sparsity/test/test_permutation_application.py
ClipGradNormTest.test_matches_pytorchMethodapex/contrib/test/clip_grad/test_clip_grad.py
make_paramsFunctionapex/contrib/test/clip_grad/test_clip_grad.py
TestCudnnGBN._test_cudnn_gbnMethodapex/contrib/test/cudnn_gbn/test_cudnn_gbn_with_two_gpus.py
GroupNormTest.verify_group_normMethodapex/contrib/test/group_norm/test_group_norm.py
OpenfoldMhaTest.test_openfold_triton_mhaMethodapex/contrib/test/openfold_triton/test_openfold_mha.py
openfold_attention_eagerFunctionapex/contrib/test/openfold_triton/test_openfold_mha.py
TestDistributedFusedAdam.test_checkpointMethodapex/contrib/test/optimizers/test_dist_adam.py
TestDistributedFusedAdam.test_matches_pytorchMethodapex/contrib/test/optimizers/test_dist_adam.py
TestDistributedFusedAdam.to_global_batchMethodapex/contrib/test/optimizers/test_dist_adam.py
TestDistributedFusedAdam.to_local_batchMethodapex/contrib/test/optimizers/test_dist_adam.py
make_modelsFunctionapex/contrib/test/optimizers/test_dist_adam.py
__get_dump_code_backends_and_dirFunctionapex/contrib/torchsched/config.py
_torchsched_codegenFunctionapex/contrib/torchsched/inductor/graph.py
transducer_joint_referenceFunctionapex/contrib/transducer/_transducer_ref.py
TransducerJoint.__init__Methodapex/contrib/transducer/transducer.py
TransducerJointFunc.forwardMethodapex/contrib/transducer/transducer.py
TransducerLoss.forwardMethodapex/contrib/transducer/transducer.py
TransducerLossFunc.forwardMethodapex/contrib/transducer/transducer.py
MLP.__init__Methodapex/mlp/mlp.py
FusedLayerNorm.__init__Methodapex/normalization/fused_layer_norm.py
FusedRMSNorm.__init__Methodapex/normalization/fused_layer_norm.py
fused_layer_norm_affineFunctionapex/normalization/fused_layer_norm.py
… +42 more
Added new API surface in @26.09
PYBIND11_MODULEFunctionapex/contrib/csrc/bottleneck/bottleneck.cpp
bottleneck_backwardFunctionapex/contrib/csrc/bottleneck/bottleneck.cpp
bottleneck_backward_grad_out1Functionapex/contrib/csrc/bottleneck/bottleneck.cpp
bottleneck_backward_grad_out1_haloFunctionapex/contrib/csrc/bottleneck/bottleneck.cpp
bottleneck_backward_grad_out1_halo_corrFunctionapex/contrib/csrc/bottleneck/bottleneck.cpp
bottleneck_backward_grad_out1_maskFunctionapex/contrib/csrc/bottleneck/bottleneck.cpp
bottleneck_backward_grad_out2Functionapex/contrib/csrc/bottleneck/bottleneck.cpp
bottleneck_backward_initFunctionapex/contrib/csrc/bottleneck/bottleneck.cpp
bottleneck_backward_restFunctionapex/contrib/csrc/bottleneck/bottleneck.cpp
bottleneck_backward_stateClassapex/contrib/csrc/bottleneck/bottleneck.cpp
bottleneck_backward_state.initMethodapex/contrib/csrc/bottleneck/bottleneck.cpp
bottleneck_backward_wgrad1Functionapex/contrib/csrc/bottleneck/bottleneck.cpp
bottleneck_backward_wgrad2Functionapex/contrib/csrc/bottleneck/bottleneck.cpp
bottleneck_backward_wgrad2_haloFunctionapex/contrib/csrc/bottleneck/bottleneck.cpp
bottleneck_backward_wgrad2_padFunctionapex/contrib/csrc/bottleneck/bottleneck.cpp
bottleneck_backward_wgrad3Functionapex/contrib/csrc/bottleneck/bottleneck.cpp
bottleneck_forwardFunctionapex/contrib/csrc/bottleneck/bottleneck.cpp
bottleneck_forward_initFunctionapex/contrib/csrc/bottleneck/bottleneck.cpp
bottleneck_forward_out1Functionapex/contrib/csrc/bottleneck/bottleneck.cpp
bottleneck_forward_out2Functionapex/contrib/csrc/bottleneck/bottleneck.cpp
bottleneck_forward_out2_haloFunctionapex/contrib/csrc/bottleneck/bottleneck.cpp
bottleneck_forward_out2_halo_corrFunctionapex/contrib/csrc/bottleneck/bottleneck.cpp
bottleneck_forward_out2_maskFunctionapex/contrib/csrc/bottleneck/bottleneck.cpp
bottleneck_forward_out2_padFunctionapex/contrib/csrc/bottleneck/bottleneck.cpp
bottleneck_forward_restFunctionapex/contrib/csrc/bottleneck/bottleneck.cpp
bottleneck_forward_statusClassapex/contrib/csrc/bottleneck/bottleneck.cpp
bottleneck_forward_status.initMethodapex/contrib/csrc/bottleneck/bottleneck.cpp
checkCudnnErrorFunctionapex/contrib/csrc/bottleneck/bottleneck.cpp
checkErrorFunctionapex/contrib/csrc/bottleneck/bottleneck.cpp
create_common_descriptorsFunctionapex/contrib/csrc/bottleneck/bottleneck.cpp
create_conv_bias_add_act_descriptorsFunctionapex/contrib/csrc/bottleneck/bottleneck.cpp
create_conv_bias_add_act_mask_descriptorsFunctionapex/contrib/csrc/bottleneck/bottleneck.cpp
create_dconv_add_descriptorsFunctionapex/contrib/csrc/bottleneck/bottleneck.cpp
create_dconv_descriptorsFunctionapex/contrib/csrc/bottleneck/bottleneck.cpp
create_dconv_mask_descriptorsFunctionapex/contrib/csrc/bottleneck/bottleneck.cpp
generateStridesFunctionapex/contrib/csrc/bottleneck/bottleneck.cpp
getConvFusionStringFunctionapex/contrib/csrc/bottleneck/bottleneck.cpp
getFwdConvDilatedFilterDimFunctionapex/contrib/csrc/bottleneck/bottleneck.cpp
getFwdConvOutputDimFunctionapex/contrib/csrc/bottleneck/bottleneck.cpp
getFwdConvPaddedImageDimFunctionapex/contrib/csrc/bottleneck/bottleneck.cpp
run_conv_add_scale_bias_activationFunctionapex/contrib/csrc/bottleneck/bottleneck.cpp
run_conv_scale_biasFunctionapex/contrib/csrc/bottleneck/bottleneck.cpp
run_conv_scale_bias_add_activationFunctionapex/contrib/csrc/bottleneck/bottleneck.cpp
run_conv_scale_bias_add_activation_maskFunctionapex/contrib/csrc/bottleneck/bottleneck.cpp
run_dconvFunctionapex/contrib/csrc/bottleneck/bottleneck.cpp
run_dconv_addFunctionapex/contrib/csrc/bottleneck/bottleneck.cpp
run_dconv_add_drelu_dscaleFunctionapex/contrib/csrc/bottleneck/bottleneck.cpp
run_dconv_drelu_dscaleFunctionapex/contrib/csrc/bottleneck/bottleneck.cpp
run_dconv_drelu_dscale_maskFunctionapex/contrib/csrc/bottleneck/bottleneck.cpp
PYBIND11_MODULEFunctionapex/contrib/csrc/conv_bias_relu/conv_bias_relu.cpp
checkCudnnErrorFunctionapex/contrib/csrc/conv_bias_relu/conv_bias_relu.cpp
checkErrorFunctionapex/contrib/csrc/conv_bias_relu/conv_bias_relu.cpp
conv_bias_backwardFunctionapex/contrib/csrc/conv_bias_relu/conv_bias_relu.cpp
conv_bias_forwardFunctionapex/contrib/csrc/conv_bias_relu/conv_bias_relu.cpp
conv_bias_mask_relu_forwardFunctionapex/contrib/csrc/conv_bias_relu/conv_bias_relu.cpp
conv_bias_relu_backwardFunctionapex/contrib/csrc/conv_bias_relu/conv_bias_relu.cpp
conv_bias_relu_forwardFunctionapex/contrib/csrc/conv_bias_relu/conv_bias_relu.cpp
conv_cscale_cbias_relu_backwardFunctionapex/contrib/csrc/conv_bias_relu/conv_bias_relu.cpp
conv_cscale_cbias_relu_forwardFunctionapex/contrib/csrc/conv_bias_relu/conv_bias_relu.cpp
generateStridesFunctionapex/contrib/csrc/conv_bias_relu/conv_bias_relu.cpp
getConvFusionStringFunctionapex/contrib/csrc/conv_bias_relu/conv_bias_relu.cpp
getFwdConvDilatedFilterDimFunctionapex/contrib/csrc/conv_bias_relu/conv_bias_relu.cpp
getFwdConvOutputDimFunctionapex/contrib/csrc/conv_bias_relu/conv_bias_relu.cpp
getFwdConvPaddedImageDimFunctionapex/contrib/csrc/conv_bias_relu/conv_bias_relu.cpp
run_conv_biasFunctionapex/contrib/csrc/conv_bias_relu/conv_bias_relu.cpp
run_conv_bias_mask_reluFunctionapex/contrib/csrc/conv_bias_relu/conv_bias_relu.cpp
run_conv_bias_reluFunctionapex/contrib/csrc/conv_bias_relu/conv_bias_relu.cpp
run_conv_cscale_cbias_reluFunctionapex/contrib/csrc/conv_bias_relu/conv_bias_relu.cpp
run_dbiasFunctionapex/contrib/csrc/conv_bias_relu/conv_bias_relu.cpp
run_dconvFunctionapex/contrib/csrc/conv_bias_relu/conv_bias_relu.cpp
run_dconv_drelu_dbiasFunctionapex/contrib/csrc/conv_bias_relu/conv_bias_relu.cpp
run_drelu_dbiasFunctionapex/contrib/csrc/conv_bias_relu/conv_bias_relu.cpp
run_drelu_dscaleFunctionapex/contrib/csrc/conv_bias_relu/conv_bias_relu.cpp
PYBIND11_MODULEFunctionapex/contrib/csrc/cudnn_gbn/cudnn_gbn.cpp
bn_typeEnumapex/contrib/csrc/cudnn_gbn/cudnn_gbn.cpp
gbn_backwardFunctionapex/contrib/csrc/cudnn_gbn/cudnn_gbn.cpp
gbn_forwardFunctionapex/contrib/csrc/cudnn_gbn/cudnn_gbn.cpp
AllowAllFunctionapex/contrib/csrc/cudnn_gbn/norm_sample.cpp
checkCudaErrorFunctionapex/contrib/csrc/cudnn_gbn/norm_sample.cpp
checkCudnnErrorFunctionapex/contrib/csrc/cudnn_gbn/norm_sample.cpp
cudaDevicePropClassapex/contrib/csrc/cudnn_gbn/norm_sample.cpp
execute_batch_norm_backwardFunctionapex/contrib/csrc/cudnn_gbn/norm_sample.cpp
execute_batch_norm_forwardFunctionapex/contrib/csrc/cudnn_gbn/norm_sample.cpp
generateStridesFunctionapex/contrib/csrc/cudnn_gbn/norm_sample.cpp
run_batch_norm_backwardFunctionapex/contrib/csrc/cudnn_gbn/norm_sample.cpp
run_batch_norm_forwardFunctionapex/contrib/csrc/cudnn_gbn/norm_sample.cpp
PYBIND11_MODULEFunctionapex/contrib/csrc/focal_loss/focal_loss_cuda.cpp
focal_loss_backwardFunctionapex/contrib/csrc/focal_loss/focal_loss_cuda.cpp
focal_loss_forwardFunctionapex/contrib/csrc/focal_loss/focal_loss_cuda.cpp
ModeEnumapex/contrib/csrc/group_norm/group_norm_nhwc.cpp
check_resultsFunctionapex/contrib/csrc/group_norm/group_norm_nhwc.cpp
group_norm_nhwc_bwd_Functionapex/contrib/csrc/group_norm/group_norm_nhwc.cpp
group_norm_nhwc_fwd_Functionapex/contrib/csrc/group_norm/group_norm_nhwc.cpp
mainFunctionapex/contrib/csrc/group_norm/group_norm_nhwc.cpp
random_dataFunctionapex/contrib/csrc/group_norm/group_norm_nhwc.cpp
unpackFunctionapex/contrib/csrc/group_norm/group_norm_nhwc.cpp
Group_norm_nhwc_bwd_paramsClassapex/contrib/csrc/group_norm/group_norm_nhwc.h
Group_norm_nhwc_fwd_paramsClassapex/contrib/csrc/group_norm/group_norm_nhwc.h
Group_sumsClassapex/contrib/csrc/group_norm/group_norm_nhwc.h
Group_sums_opClassapex/contrib/csrc/group_norm/group_norm_nhwc.h
… +344 more