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 {

    template <Device D, typename T>
    void ReduceAll::compute(const StorageView& /*input*/, StorageView& /*output*/) const {
      throw std::runtime_error("reduce all is not applied for the cpu");
    }

    template <Device D, typename T>
    void GatherAll::compute(const StorageView& /*input*/, StorageView& /*output*/) const {
      throw std::runtime_error("gather all is not applied for the cpu");
    }
    #define DECLARE_IMPL(T)                                                 \
        template void ReduceAll::compute<Device::CPU, T>(const StorageView&, \
                                                          StorageView&) const; \
        template void GatherAll::compute<Device::CPU, T>(const StorageView&, \
                                                          StorageView&) const;
    DECLARE_ALL_TYPES(DECLARE_IMPL)
  }
}