Skip to content

Commit 2706059

Browse files
authored
[EM] Remove warning for ATS. (#12069)
1 parent a7eeece commit 2706059

File tree

1 file changed

+1
-3
lines changed

1 file changed

+1
-3
lines changed

src/data/ellpack_page_source.h

Lines changed: 1 addition & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -187,10 +187,8 @@ class EllpackFormatPolicy {
187187
public:
188188
EllpackFormatPolicy() {
189189
StringView msg{" The overhead of iterating through external memory might be significant."};
190-
if (!has_hmm_) {
190+
if (!(has_hmm_ || curt::SupportsAts())) {
191191
LOG(WARNING) << "CUDA heterogeneous memory management is not available." << msg;
192-
} else if (!curt::SupportsAts()) {
193-
LOG(WARNING) << "CUDA address translation service is not available." << msg;
194192
}
195193
if (!(GlobalConfigThreadLocalStore::Get()->use_rmm ||
196194
GlobalConfigThreadLocalStore::Get()->use_cuda_async_pool)) {

0 commit comments

Comments
 (0)