marian-nmt · rhenry-nv · Oct 17, 2020 · Oct 19, 2020 · Oct 21, 2020 · Oct 22, 2020
diff --git a/.gitmodules b/.gitmodules
@@ -20,3 +20,6 @@
 [submodule "src/3rd_party/simple-websocket-server"]
 	path = src/3rd_party/simple-websocket-server
 	url = https://github.com/marian-nmt/Simple-WebSocket-Server
+[submodule "src/3rd_party/cub"]
+	path = src/3rd_party/cub
+	url = https://github.com/NVIDIA/cub
diff --git a/CHANGELOG.md b/CHANGELOG.md
@@ -9,6 +9,9 @@ and this project adheres to [Semantic Versioning](http://semver.org/spec/v2.0.0.
 ## [Unreleased]
 
 ### Added
+
+- Includes cub as a dependency
+- Replaces the topK implementation in nth_element.cu and topk.cu
 - Local/global sharding with MPI training via `--sharding local`
 - fp16 support for factors.
 - Correct training with fp16 via `--fp16`. 

diff --git a/LICENSE.md b/LICENSE.md
@@ -3,6 +3,8 @@ MIT License
 Copyright (c) 2016 Marcin Junczys-Dowmunt, the University of Edinburgh, Adam 
 Mickiewicz University
 
+Copyright (c) 2020 NVIDIA Corporation
+
 Permission is hereby granted, free of charge, to any person obtaining a copy
 of this software and associated documentation files (the "Software"), to deal
 in the Software without restriction, including without limitation the rights

diff --git a/src/3rd_party/cub b/src/3rd_party/cub
diff --git a/src/3rd_party/topk.cuh b/src/3rd_party/topk.cuh
diff --git a/src/CMakeLists.txt b/src/CMakeLists.txt
@@ -1,3 +1,5 @@
+add_definitions(-DCUB_IGNORE_DEPRECATED_CPP_DIALECT=1)
+add_definitions(-DTHRUST_IGNORE_DEPRECATED_CPP_DIALECT=1)
 add_subdirectory(3rd_party)
 
 include_directories(.)