ct2rs 0.9.19

Rust bindings for OpenNMT/CTranslate2
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23

#include "ctranslate2/ops/nccl_ops.h"
#include "dispatch.h"

namespace ctranslate2 {
  namespace ops {

    ReduceAll::ReduceAll(ReduceAll::RED_OP op)
      : _reduce_op(op) {
    }

    void ReduceAll::operator()(const StorageView& input, StorageView& output) const {
      PROFILE("ReduceAll");
      DEVICE_AND_TYPE_DISPATCH(input.device(), input.dtype(), (compute<D, T>(input, output)));
    }

    GatherAll::GatherAll() = default;

    void GatherAll::operator()(const StorageView& input, StorageView& output) const {
      PROFILE("ReduceAll");
      DEVICE_AND_TYPE_DISPATCH(input.device(), input.dtype(), (compute<D, T>(input, output)));
    }
  }
}