fix fp32 (#79)

Co-authored-by: Dmitry Razdoburdin <>

Author: razdoburdin

plugin/sycl/common/linalg_op.h

@@ -103,7 +103,7 @@ bool Validate(DeviceOrd device, TensorView<T, D> t, Fn&& fn) {
 namespace linalg {
 template <typename T, int32_t D, typename Fn>
 void ElementWiseKernel(Context const* ctx, TensorView<T, D> t, Fn&& fn) {
-  if (ctx->IsSycl()) {
+  if (t.Device().IsSycl()) {
     sycl::linalg::ElementWiseKernel(t, fn);
   } else {
     ElementWiseKernelHost(t, ctx->Threads(), fn);

src/objective/multiclass_obj.cu

@@ -110,7 +110,7 @@ class SoftmaxMultiClassObj : public ObjFunction {
           << "Number of weights should be equal to number of data points.";
     }
     info.weights_.SetDevice(device);
-    auto weights = common::MakeOptionalWeights(this->ctx_->Device(), info.weights_);
+    auto weights = common::MakeOptionalWeights(device, info.weights_);
 
     preds.SetDevice(device);
     auto predt = linalg::MakeTensorView(this->ctx_, &preds, n_samples, n_classes);