Switch branches/tags
JoelMarcey-patch-1 Jorghi12-PytorchROCmRemove_Deps Jorghi12-patch-2 Jorghi12-patch-6 Jorghi12-transpiler PyRocm_removing_deps SsnL-patch-1 SsnL-patch-2 SsnL-patch-3 SsnL-patch-4 SsnL-patch-5 a5d7abe add-issue-templates always_scriptmodule anderspapitto-patch-1 batch_mm_t build_fixes build_incremental_fix cache_vars caffe2-docker-image-rebuild circleci_all_commits circleci_credentials circleci_disable_all_jobs circleci_fetch_depth circleci_fix_git circleci_flaky_tests circleci_generic_kernel circleci_test circleci_timestamp circleci cleanup_linkage cpp cudastaticfixes docker/rocm-1.8.2 docker/rocm-update export-D9526248 export-D9526650 export-D9526737 export-D9539945 export-D9540025 export-D9545248 export-D9545704 export-D9557315 export-D9561478 export-D9561802 export-D9562197 export-D9562312 export-D9562467 export-D9563464 export-D9563753 export-D9564206 export-D9564516 export-D9578397 export-D9578398 export-D9578399 export-D9578734 export-D9579371 export-D9581560 export-D9583630 export-D9583699 export-D9585627 export-D9613800 export-D9613897 export-D9614321 export-D9623916 export-D9631619 export-D9634750 export-D9634904 export-D9635105 export-D9635292 export-D9644700 export-D9644899 export-D9646190 export-D9648570 export-D9648830 export-D9652088 export-D9652089 export-D9654871 export-D9656548 export-D9657449 export-D9663476 export-D9666612 export-D9670493 export-D9676205 export-D9694097 export-D9694326 export-D9694327 export-D9694918 export-D9697878 export-D9724805 export-D9727134 export-D9727532 export-D9728631 export-D9731326 export-D9756666 export-D9757935 export-D9763422 export-D9763423 export-D9763424 export-D9771708 export-D9775191 export-D9778042 export-D9778043 export-D9779821 export-D9790187 export-D9806425 export-D9810823 export-D9811028 export-D9813544 export-D9813742 export-D9814536 export-D9823457 export-D9826209 export-D9830460 export-D9831230 export-D9831384 export-D9833361 export-D9841660 export-D9847835 export-D9847859 export-D9882726 export-D9884177 export-D9884563 export-D9889990 export-D9924348 export-D9967509 export-D9968041 export-D9968320 export-D9977058 export-D9977505 export-D9977654 export-D9979976 export-D9980641 export-D9995559 export-D9995633 export-D9996898 export-D10001033 export-D10022853 export-D10024439 export-D10024467 export-D10024485 export-D10024554 export-D10026392 export-D10030556 export-D10030819 export-D10031072 export-D10032707 export-D10033396 export-D10034589 export-D10037265 export-D10050781 export-D10050859 export-D10050905 export-D10051005 export-D10051012 export-D10051078 export-D10051079 export-D10051126 export-D10051202 export-D10051298 export-D10051365 export-D10051424 export-D10052523 export-D10069839 export-D10073519 export-D10111759 export-D10134083 export-D10134319 export-D10139933 export-D10139934 export-D10139935 export-D10150834 export-D10184116 export-D10184117 export-D10200448 export-D10204135 export-D10207890 export-D10209620 export-D10216315 export-D10222739 export-D10227820 export-D10229684 export-D10232118 export-D10232147 export-D10232154 export-D10249293 export-D10251907 export-D10255651 export-D10359443 export-D10371541 export-D10379903 export-D10380678 export-D10392295 export-D10400927 export-D10404407 export-D10415069 export-D10415430 export-D10416051 export-D10419671 export-D10421896 export-D10450290 export-D10454455 export-D10457671 export-D10467239 export-D10467556 export-D10469310 export-D10469960 export-D10476220 export-D10476225 export-D10476226 export-D10476232 export-D10476235 export-D10488399 export-D10492071 export-D10492507 export-D10496244 export-D10513246 export-D10518499 export-D10518929 export-D10520295 export-D10520421 export-D10528061 export-D10853224 export-D10855883 export-D10858024 export-D11669870 export-D12143282 export-D12832080 export-D12848855 export-D12849620 export-D12850690 export-D12850691 export-D12850833 export-D12873145 export-D12874357 export-D12894385 export-D12894386 export-D12912235 export-D12912237 export-D12912238 export-D12912239 export-D12912240 export-D12912241 export-D12912242 export-D12934074 export-D12936031 export-D12937090 export-D12937091 export-D12964886 export-D12985774 export-D13009482 export-D13011878 export-D13015236 export-D13015239 export-D13024368 export-D13025313 export-D13036478 export-D13046201 export-D13046500 export-D13046722 export-D13047468 export-D13053648 export-D13056152 export-D13062526 export-D13062564 export-D13062604 export-D13062631 export-D13062649 export-D13062706 export-D13066808 export-D13081602 export-D13081603 export-D13081604 export-D13081605 export-D13081606 export-D13081607 export-D13081608 export-D13081609 export-D13081610 export-D13104693 export-D13104694 export-D13105166 export-D13111509 export-D13111712 export-D13111781 export-D13112081 export-D13112298 export-D13113129 export-D13119624 export-D13121531 export-D13128077 export-D13128977 export-D13131338 export-D13141949 export-D13145293 export-D13156470 export-D13156471 export-D13156472 export-D13158474 export-D13158475 export-D13205022 export-D13218540 export-D13221302 export-D13223125 export-D13223126 export-D13223904 export-D13224015 export-D13235001 export-D13241355 export-D13241401 export-D13257847 export-D13258252 export-D13258512 export-D13258513 export-D13266063 export-D13267832 export-D13271560 export-D13272227 export-D13277246 export-D13277567 export-D13283492 export-D13283493 export-D13283494 export-D13283495 export-D13283496 export-D13283497 export-D13285370 export-D13287688 export-D13288655 export-D13304398 export-D13316078 export-D13318594 export-D13318596 export-D13318644 export-D13318645 export-D13336841 export-D13336842 export-D13336843 export-D13336856 export-D13348039 export-D13348040 export-D13348041 export-D13348042 export-D13348044 export-D13349163 export-D13349164 export-D13365817 export-D13401512 export-D13425628 export-D13427897 export-D13432800 export-D13432922 export-D13441008 export-D13445571 export-D13451550 export-D13461640 export-D13462124 export-D13463673 export-D13473791 ext_test_fix ezyang-patch-1 ezyang-patch-2 ezyang-patch-3 ezyang-patch-4 ezyang/retry-type-id-core ezyang/rocm-docker-update fast_dp fb-config fbsync fix_inplace_double_backward gh/ezyang/1/base gh/ezyang/1/head gh/ezyang/1/orig gh/ezyang/2/base gh/ezyang/2/head gh/ezyang/2/orig gh/ezyang/3/base gh/ezyang/3/head gh/ezyang/3/orig gloo_dedup halfconv jerryzh168-patch-1-1 jerryzh168-patch-1 jit_frontend known-good legacy_remove magmatestfix master merge_variable_tensor mkl_set_dynamic nccl_fix new_symbolic_diff nn_c_port oanderso/test random_device readme_fix remove_time scalar_type simple_engine soumith-patch-1 ssnl-9348 suo/annotations suo/clang-format suo/dce2 suo/fix-expect suo/graph-equals suo/ir-parser suo/mm suo/parser suo/schematize suo/slicer tensor-merge tensorimpl_autogradmeta tensorimpl_3_variable_functions tensorimpl_4_AutogradMetaInterface test10 testing/full-caffe2 tmp_enable_scalars v0.4.1 v1.0.0 weak_tensor weak_tracing win_py2.7 windows_jit_error windows_jit_error_12378
Nothing to show
Find file History
rohithkrn and facebook-github-bot FP16MomentumSGDUpdate Op fix and enable for ROCm (#15150)
Summary:
1. Fix a bug in FP16MomentumSGDUpdate operator
2. Enable operator for ROCm
Pull Request resolved: #15150

Differential Revision: D13473145

Pulled By: bddppq

fbshipit-source-id: 4c5c5f30cb9bba658e3639dbe193fa08a304d306
Latest commit 763b995 Dec 14, 2018
Permalink
Type Name Latest commit message Commit time
..
Failed to load latest commit information.
docs Add __init__.py so files get picked up on install (#14898) Dec 7, 2018
examples Add __init__.py so files get picked up on install (#14898) Dec 7, 2018
helpers [Caffe2] Add full impl of GroupNorm (#7058) Apr 29, 2018
ideep Implementation of ChannelShuffle Op for MKLDNN (#15106) Dec 13, 2018
layers Resubmit: Set the correct engine name for position weighted pooling w… Nov 27, 2018
mint re-enable copy of python files, but be careful that the copy is only … ( Dec 12, 2018
mkl Add __init__.py so files get picked up on install (#14898) Dec 7, 2018
modeling diagnose option: get_entry to print a whole row (#11308) Sep 7, 2018
models Unify cuda and hip device types in Caffe2 python front end (#14221) Nov 29, 2018
onnx Run ONNX cuda backend test cases via ROCm Dec 13, 2018
operator_test FP16MomentumSGDUpdate Op fix and enable for ROCm (#15150) Dec 15, 2018
predictor Caffe2: Fix for creating entries of external_input in predic_net (#12979 Nov 16, 2018
rnn Unify cuda and hip device types in Caffe2 python front end (#14221) Nov 29, 2018
serialized_test docs: minor spelling tweaks Dec 13, 2018
test Back out "Revert D13043261: [caffe2] Task graph and task future abstr… Dec 11, 2018
trt USE_TENSORRT support and TensorRT 5 compatibility Dec 7, 2018
CMakeLists.txt Fix CMakeLists.txt for Int8 python bindings (#15047) Dec 11, 2018
__init__.py caffe2::DeviceType -> at::DeviceType (#11254) Sep 5, 2018
_import_c_extension.py Remove linker and dlopen flags that allowed undefined symbols in rocm… Dec 12, 2018
allcompare_test.py Remove Apache headers from source. Mar 27, 2018
attention.py [Caffe2] Update elementwise ops to support numpy style boradcast (#8070) Jun 5, 2018
benchmark_generator.py Remove Apache headers from source. Mar 27, 2018
binarysize.py Remove Apache headers from source. Mar 27, 2018
brew.py [Caffe2] Changes done inside Facebook (#6378) Apr 11, 2018
brew_test.py Move tanh function to math (#9328) Jul 11, 2018
build.py Remove Apache headers from source. Mar 27, 2018
cached_reader.py Update from Facebook (#8887) Jun 26, 2018
caffe_translator.py Fix bug in caffe_translator tool (#10056) Oct 11, 2018
caffe_translator_test.py Fix skip logic in caffe_translator_test.py (#13627) Nov 16, 2018
checkpoint.py Create class constant for string literal 'blob_names' Aug 25, 2018
checkpoint_test.py Revert D9566744: [New Checkpoint] Kill the dummy TaskOutput when task… Sep 1, 2018
cnn.py Unify cuda and hip device types in Caffe2 python front end (#14221) Nov 29, 2018
compatibility.py migrating deprecated calls without abc module for containers (#11515) Sep 13, 2018
context.py Resolve name conflict of ContextManager (#7244) Jun 22, 2018
context_test.py Remove Apache headers from source. Mar 27, 2018
control.py Remove Apache headers from source. Mar 27, 2018
control_ops_grad.py fix auto grad summing for IfOp where intermediate output needs renami… Dec 9, 2018
control_ops_grad_test.py fix auto grad summing for IfOp where intermediate output needs renami… Dec 9, 2018
control_ops_util.py Remove Apache headers from source. Mar 27, 2018
control_test.py Remove Apache headers from source. Mar 27, 2018
convert.py New serialization format (#12384) Oct 16, 2018
convert_test.py New serialization format (#12384) Oct 16, 2018
convnet_benchmarks.py Remove Apache headers from source. Mar 27, 2018
convnet_benchmarks_test.py Remove Apache headers from source. Mar 27, 2018
core.py fix auto grad summing for IfOp where intermediate output needs renami… Dec 9, 2018
core_gradients_test.py Unify cuda and hip device types in Caffe2 python front end (#14221) Nov 29, 2018
core_test.py Unify cuda and hip device types in Caffe2 python front end (#14221) Nov 29, 2018
crf.py Productionize CRF layer in PyText (#10362) Aug 22, 2018
crf_predict.py Move crf in caffe2 from fb to oss (#12200) Oct 2, 2018
crf_viterbi_test.py Move crf in caffe2 from fb to oss (#12200) Oct 2, 2018
data_parallel_model.py Unify cuda and hip device types in Caffe2 python front end (#14221) Nov 29, 2018
data_parallel_model_test.py Unify cuda and hip device types in Caffe2 python front end (#14221) Nov 29, 2018
data_workers.py Fixed log message (#10874) Sep 5, 2018
data_workers_test.py Remove Apache headers from source. Mar 27, 2018
dataio.py Fixing stop condition on composite reader (#9888) Aug 20, 2018
dataio_test.py Update from Facebook (#8887) Jun 26, 2018
dataset.py Update from facebook (#7855) May 29, 2018
db_file_reader.py Update from Facebook (#8887) Jun 26, 2018
db_test.py Remove Apache headers from source. Mar 27, 2018
device_checker.py Update from facebook (#7451) May 11, 2018
dlpack.h Upgrade DLPack Nov 12, 2018
dyndep.py Remove Apache headers from source. Mar 27, 2018
embedding_generation_benchmark.py Remove Apache headers from source. Mar 27, 2018
experiment_util.py Remove Apache headers from source. Mar 27, 2018
extension_loader.py Completely remove build_aten and use_aten (#10469) Aug 21, 2018
functional.py Caffe2 Functional enforcing inplace output (#10797) Aug 24, 2018
functional_test.py Add support for specifying device_option in Functional (#9619) Jul 24, 2018
fused_8bit_rowwise_conversion_ops_test.py Remove Apache headers from source. Mar 27, 2018
gradient_check_test.py Unify cuda and hip device types in Caffe2 python front end (#14221) Nov 29, 2018
gradient_checker.py make the variable declaration closer to usage Oct 12, 2018
gru_cell.py Remove Apache headers from source. Mar 27, 2018
hip_test_util.py Make CUDNN an alias of MIOPEN for HIP ops (#12278) Oct 25, 2018
hsm_util.py Remove Apache headers from source. Mar 27, 2018
hypothesis_test.py Remove unsafecoalesce op (#12897) Oct 22, 2018
hypothesis_test_util.py Unify cuda and hip device types in Caffe2 python front end (#14221) Nov 29, 2018
ideep_test_util.py [feature request] [Caffe2] Enable MKLDNN support for inference (#6699) Apr 23, 2018
layer_model_helper.py parallize the dense part in event models Aug 23, 2018
layer_model_instantiator.py [caffe2] Fbcode to GitHub sync (#6208) Apr 2, 2018
layer_parameter_sharing_test.py Clean up a couple of items in the C2 test scaffolding (WIP) (#7847) Nov 7, 2018
layer_test_util.py Remove Apache headers from source. Mar 27, 2018
layers_test.py Add Recency Weighted into SparseLookup (#14291) Nov 24, 2018
lengths_reducer_fused_8bit_rowwise_ops_test.py Remove Apache headers from source. Mar 27, 2018
lengths_reducer_rowwise_8bit_ops_test.py Remove Apache headers from source. Mar 27, 2018
lstm_benchmark.py Unify cuda and hip device types in Caffe2 python front end (#14221) Nov 29, 2018
memonger.py Remove Apache headers from source. Mar 27, 2018
memonger_test.py Unify cuda and hip device types in Caffe2 python front end (#14221) Nov 29, 2018
mkl_test_util.py Remove Apache headers from source. Mar 27, 2018
model_device_test.py Unify cuda and hip device types in Caffe2 python front end (#14221) Nov 29, 2018
model_helper.py Add error type to raise statement Dec 12, 2018
model_helper_test.py keep net type info when generating model complete net (#11032) Sep 5, 2018
modifier_context.py Remove Apache headers from source. Mar 27, 2018
mpi_python.cc Remove Apache headers from source. Mar 27, 2018
muji.py Unify cuda and hip device types in Caffe2 python front end (#14221) Nov 29, 2018
muji_test.py Unify cuda and hip device types in Caffe2 python front end (#14221) Nov 29, 2018
net_builder.py Update from Facebook (#6692) Apr 18, 2018
net_builder_test.py Remove Apache headers from source. Mar 27, 2018
net_drawer.py Remove Apache headers from source. Mar 27, 2018
net_printer.py Rename cuda_gpu_id to device_id in DeviceOption (#12456) Oct 9, 2018
net_printer_test.py Remove Apache headers from source. Mar 27, 2018
nomnigraph.py createUniqueDataNode Oct 31, 2018
nomnigraph_test.py nomnigraph - support subgraph visualization (#13795) Nov 16, 2018
nomnigraph_transformations.py Add transpose network pass (#13437) Nov 1, 2018
nomnigraph_transformations_test.py Add transpose network pass (#13437) Nov 1, 2018
normalizer.py Enable alternative LayerNorm impl in FisherGan (#12178) Oct 12, 2018
normalizer_context.py Update from Facebook (#8887) Jun 26, 2018
normalizer_test.py Update from Facebook (#8887) Jun 26, 2018
numa_benchmark.py Revert D13205604: Move numa.{h, cc} to c10/util Dec 7, 2018
numa_test.py Move numa.{h, cc} to c10/util (#15024) Dec 12, 2018
observer_test.py Fix RNN scoping situation Feb 8, 2018
optimizer.py Unify cuda and hip device types in Caffe2 python front end (#14221) Nov 29, 2018
optimizer_context.py Remove Apache headers from source. Mar 27, 2018
optimizer_test.py Unify cuda and hip device types in Caffe2 python front end (#14221) Nov 29, 2018
optimizer_test_util.py Unify cuda and hip device types in Caffe2 python front end (#14221) Nov 29, 2018
parallel_workers.py Update from facebook (#7696) May 20, 2018
parallel_workers_test.py Remove Apache headers from source. Mar 27, 2018
parallelize_bmuf_distributed_test.py Unify cuda and hip device types in Caffe2 python front end (#14221) Nov 29, 2018
pipeline.py SNNTest with Data Preproc Service (#11707) Sep 18, 2018
pipeline_test.py Remove Apache headers from source. Mar 27, 2018
predictor_constants.py Remove Apache headers from source. Mar 27, 2018
pybind_state.cc caffe2/caffe2/contrib/script (#15007) Dec 10, 2018
pybind_state.h Add inplace FeedTensor for python frontend (#14512) Dec 4, 2018
pybind_state_dlpack.cc Upgrade DLPack Nov 12, 2018
pybind_state_dlpack.h Upgrade DLPack Nov 12, 2018
pybind_state_gpu.cc Renaming dims() to sizes() (caffe2/caffe2) - 4/4 Oct 24, 2018
pybind_state_hip.cc Change hip filename extension to .hip (#14036) Nov 16, 2018
pybind_state_ideep.cc Add inplace FeedTensor for python frontend (#14512) Dec 4, 2018
pybind_state_int8.cc Renaming meta() to dtype() - 2/2 (#13334) Oct 31, 2018
pybind_state_nomni.cc nomnigraph - support subgraph visualization (#13795) Nov 16, 2018
pybind_state_registry.cc Move registry fully to c10 (#12077) Sep 27, 2018
pybind_state_registry.h Move registry fully to c10 (#12077) Sep 27, 2018
python_op_test.py Clean up a couple of items in the C2 test scaffolding (WIP) (#7847) Nov 7, 2018
queue_util.py Remove Apache headers from source. Mar 27, 2018
record_queue.py Remove Apache headers from source. Mar 27, 2018
recurrent.py Remove Apache headers from source. Mar 27, 2018
regularizer.py Add GroupL1Norm regularizer (#9115) Jul 6, 2018
regularizer_context.py Update from Facebook (#8887) Jun 26, 2018
regularizer_test.py Add GroupL1Norm regularizer (#9115) Jul 6, 2018
rnn_cell.py Unify cuda and hip device types in Caffe2 python front end (#14221) Nov 29, 2018
schema.py Add util function from core type to dtype (#10716) Aug 21, 2018
schema_test.py Add util function from core type to dtype (#10716) Aug 21, 2018
scope.py Add EmptyNameScope to allow you jump out from current scope. (#14631) Dec 12, 2018
scope_test.py Add EmptyNameScope to allow you jump out from current scope. (#14631) Dec 12, 2018
session.py Remove Apache headers from source. Mar 27, 2018
session_test.py Remove Apache headers from source. Mar 27, 2018
sparse_to_dense_mask_test.py Remove Apache headers from source. Mar 27, 2018
sparse_to_dense_test.py Remove Apache headers from source. Mar 27, 2018
task.py Allowing TaskGroups to carry remote nets (#14342) Nov 27, 2018
test_util.py Start unittesting our main observer (#15191) Dec 15, 2018
text_file_reader.py Remove Apache headers from source. Mar 27, 2018
timeout_guard.py Remove Apache headers from source. Mar 27, 2018
toy_regression_test.py Enable junk fill for the default CPU allocator (#13377) Nov 8, 2018
transformations.py Enable Conv fusion optimizations in optimizeForIdeep (#9255) Jul 17, 2018
transformations_test.py nomnigraph - easy - some code cleanup for transformations_test (#12101) Oct 1, 2018
tt_core.py Remove Apache headers from source. Mar 27, 2018
tt_core_test.py Remove Apache headers from source. Mar 27, 2018
utils.py Unify cuda and hip device types in Caffe2 python front end (#14221) Nov 29, 2018
utils_test.py Convert Arguments to dictionary (#13436) Nov 1, 2018
visualize.py Remove Apache headers from source. Mar 27, 2018
workspace.py Unify cuda and hip device types in Caffe2 python front end (#14221) Nov 29, 2018
workspace_test.py Unify cuda and hip device types in Caffe2 python front end (#14221) Nov 29, 2018