| 234 | |
| 235 | CompNodeEnv::AtlasEnv::InitStatus CompNodeEnv::AtlasEnv::init_status; |
| 236 | void CompNodeEnv::init_atlas(CompNode comp_node, const AtlasEnv& env) { |
| 237 | m_comp_node = comp_node; |
| 238 | m_atlas_env = env; |
| 239 | m_property.type = DeviceType::ATLAS; |
| 240 | m_property.mem_alignment = 64; |
| 241 | |
| 242 | m_atlas_env.activate(); |
| 243 | MGB_ATLAS_CHECK(aclrtCreateStream(&m_atlas_env.stream)); |
| 244 | m_user_data_container = std::make_unique<UserDataContainer>(); |
| 245 | mgb_assert( |
| 246 | m_property.mem_alignment == |
| 247 | MegDNNHandle::get(*this).handle()->alignment_requirement()); |
| 248 | } |
| 249 | #endif |
| 250 | |
| 251 | #if MGB_ROCM |
no test coverage detected