How to get the wgpu buffer of a tensor on the new 0.17 release?

[paulmaksimovich]

Previously you could:

pub type JitWgpuBackend = JitBackend<WgpuRuntime, f32, i32, u32>;
let tensor: Tensor<JitWgpuBackend, 3> = ....
let jit_tensor = tensor.clone().into_primitive().tensor();

let resource = jit_tensor
    .client
    .get_resource(jit_tensor.handle.binding());

let gpu_buffer = &resource.resource().buffer;

Is there still a way to access the buffer?

[paulmaksimovich]

Ah figured out my own answer, using the CubeBackend with WgpuRuntime:

pub type Backend = CubeBackend<WgpuRuntime, f32, i32, u32>;
let tensor: Tensor<JitWgpuBackend, 3> = ....
let jit_tensor = tensor.clone().into_primitive().tensor();

let resource = jit_tensor
    .client
    .get_resource(jit_tensor.handle.binding());
let gpu_buffer = &resource.resource().buffer();