I want to generate SampleUniform Tensor on GPU backend with TVM, but I find 
that it has not been realized.

  void 
[SampleUniform](http://tracking.discuss.tvm.ai/tracking/click?d=iGLiP2fCNomqXQ9RM8e6LMDoFz224tfuxlz6r25Bj-dc7FRkVni2rhU9AzgtqLrhcb6rmRLv20UbL-4ozwo7z27qTynZiIWO_klPc6wII-PV8RoZiTcG92Ho8gOdKhuZ-dXRWHp3wk3tLo3Ih89MCiDRmwhXQBvYEHAITpzW8nx-7Fc3JXsbqxOynrkOzRz0Sw2)(DLTensor*
 data, float low, float high) {
    CHECK_GT(high, low) << "high must be bigger than low";
    CHECK(data->strides == nullptr);

    DLDataType dtype = data->dtype;
    int64_t size = 1;
    for (int i = 0; i < data->ndim; ++i) {
      size *= data->shape[i];
    }

    CHECK(dtype.code == kDLFloat && dtype.bits == 32 && dtype.lanes == 1);

    if (data->ctx.device_type == kDLCPU) {
      std::uniform_real_distribution<float> uniform_dist(low, high);
      std::generate_n(static_cast<float*>(data->data), size, [&] () {
        return uniform_dist(rnd_engine_);
      });
    } else {
      LOG(FATAL) << "Do not support random.uniform on this device yet";
    }
  }

Have anyone come across similar problems?
Could you give me some suggestions on how to relaize the ELSE 
part(data->ctx.device_type == kDLGPU)? 
Thanks a lot
@nhynes
@tqchen





---
[Visit 
Topic](http://tracking.discuss.tvm.ai/tracking/click?d=1b86q3_wBAJJtspDG7F7ZSGafjq8y8xQEzR427UQ0VqvNls2v7Xj8rTQR06JUIRo7Pjw0gZxfMfa3shtUcf2msENLbtn5LKz4VNtt8XKw9IOWHqTmRP9jfeLdfBj4pG4bg4a1s0FaZ4e5IU1WoTiDk3heQMXhLT_gHkAxdLWPH15DJwoGlZ9MIc08GfP_95JKw2)
 to respond.

You are receiving this because you enabled mailing list mode.

To unsubscribe from these emails, [click 
here](http://tracking.discuss.tvm.ai/tracking/click?d=7cFgOaAA4XIBVlVKt_oyC07uihTjg4Q6cjeBRNRTiPo5aCgNxedB0LbKVfiV5DahyO7_OrIQrbTpyqDI9vRD2NDCiSnX-f_dwQzZEXuI6RhFkxA_NJxm0jaLiyoLSgfWyvVmak6m2vNvdVtI5iFKS2D_9MZwH-maQrYFcCMZSn5AKi8dOQ93s3vaau9tl9nWntfIbmheWg-W4xEYUBrPXwJODQAxhtlYtSbB34pgj27q0).

Tianqi Chen, UW, Seattle, WA, 98105, United States
http://tracking.discuss.tvm.ai/tracking/unsubscribe?msgid=Zm9q2vi9PX_z42oqPJKSjg2

Reply via email to