Hui Liu

SketchBatchNumElements minor bugs

1

1) In xgboost/blob/master/src/common/hist_util.cu, SketchBatchNumElements, auto avail = dh::AvailableMemory(device) * 0.8; this is a **double**, and this function returns size_t. return value, return **std::min**(sketch_batch_num_elements, kIntMax); 2) In xgboost/blob/master/tests/cpp/common/test_hist_util.cu, TEST(HistUtil, SketchBatchNumElements), auto...

## Overview This PR adds a few files under cext/ folder and enables AMD GPUs using ROCm. The setup.py has been modified to detect ROCm and to compile HIP codes.

fix tmp_out and exp_sums dimensions

1

The first dimension of tmp_out and exp_sums is inferred from block_tables.size(0), which may be different from query.shape(0). The later can be much larger than block_tables.size(0), which may cause OOM. This...

Add MI300X Spec

2

A MI300X spec has been provided. Thanks.