8000 supports `device` argument for factory methods · Issue #175 · NVIDIA/Fuser · GitHub
[go: up one dir, main page]
More Web Proxy on the site http://driver.im/
Skip to content
supports device argument for factory methods #175
New issue
New issue
Closed
Closed
supports device argument for factory methods#175
Assignees
jacobhinkle
@jjsjann123

Description

@jjsjann123
jjsjann123
opened on Apr 17, 2023

nvfuser should support device argument.

We could have factory methods in fusion that doesn't have a tensor input, so device argument would be necessary to determine GPU device for kernel generation.

e.g. in cpp tests

Fuser/test/test_gpu_tensor_factories.cpp

Lines 51 to 56 in 44e99fe

auto out_tv = full({size}, fill_val1, aten_to_data_type(dtype));
fusion->addOutput(out_tv);
out_tv = full({size, size}, fill_val2, aten_to_data_type(dtype));
fusion->addOutput(out_tv);
out_tv = full_like(out_tv, fill_val3);
fusion->addOutput(out_tv);

Note that these arguments should also be plumbed all the way through to python API.

Metadata

Metadata

Assignees

Labels

No labels
No labels

Type

No type

Projects

No projects

Milestone

No milestone

Relationships

None yet

Development

No branches or pull requests

Issue actions
    0