Aligment not passed to TVMBackendAllocWorkspace?

I have something like this:

  // attr [compute] storage_alignment = 128
  let compute = TVMBackendAllocWorkspace(14, dev_id, (uint64)4, 0, 32)

It seems like TVMBackendAllocWorkspace should take alignment as an argument, or otherwise there is no way to enforce it. Should it be added to the API?

1 Like