| 113 | } |
| 114 | |
| 115 | int TensorImplDft::get_device_id() const { |
| 116 | if (is_host()) { |
| 117 | return m_host_tensor->comp_node().locator().device; |
| 118 | } else { |
| 119 | return m_dev_tensor->comp_node().locator().device; |
| 120 | } |
| 121 | } |
| 122 | |
| 123 | bool TensorImplDft::is_pinned_host() const { |
| 124 | return is_host() && get_device_from_locator(m_host_tensor->comp_node().locator()) != |