Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Cuda fuser build fix #6

Closed
wants to merge 442 commits into from
Closed
Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
442 commits
Select commit Hold shift + click to select a range
71a74de
clang-format
bertmaher Jan 13, 2020
bdf91f3
A functinoal reference chck for schedule tests.
zheng-xq Jan 13, 2020
0a906e8
clang-format
zheng-xq Jan 13, 2020
af57da7
Add support for Float immediates.
Jan 13, 2020
92d550d
Get absolute path for ASMJIT_DIR (#24)
bertmaher Jan 13, 2020
658ded9
Silence deprecation warnings from LLVM
bertmaher Jan 13, 2020
ef77c6f
Include legacy PassManager for debug printing
bertmaher Jan 13, 2020
cd7928f
Set code model to medium to avoid indirect jumps in generated asm
bertmaher Jan 13, 2020
7ed7516
Fix argument type of input float buffers
bertmaher Jan 13, 2020
54bb328
Add support for Casts in LLVM codegen.
Jan 13, 2020
a0bc9bf
Add a complete tensor+lower+llvm test
zheng-xq Jan 14, 2020
58738ed
Enable the failing test
zheng-xq Jan 14, 2020
54ca59b
Enable export of compile_commands.json.
Jan 13, 2020
8fcf9a7
Floating point arithmetic
bertmaher Jan 14, 2020
9412adb
Test fp32 mul using compute expr
bertmaher Jan 14, 2020
9f26b7f
Broadcast add test using compute expr
bertmaher Jan 14, 2020
cce0cd3
Update to LLVM 9
Jan 14, 2020
e7c6855
Implementation of Broadcast for LLVM.
Jan 14, 2020
320cdf5
Fix fusion container/guard.
csarofeen Feb 1, 2020
bace796
Move tests for GPU fusion.
csarofeen Feb 1, 2020
1d08f9c
Remove region. Add simple mutator.
csarofeen Feb 3, 2020
e1f1827
Comment ir.h/.cpp
csarofeen Feb 4, 2020
ea1d04e
Comment fusion.h
csarofeen Feb 4, 2020
24285b8
Comment mutator and visitor.
csarofeen Feb 4, 2020
fa9f84b
Comment arith.h
csarofeen Feb 4, 2020
3fa3337
it builds!
jjsjann123 Jan 17, 2020
56d06a8
codegen compiled and working (with hardcoded stuff)
jjsjann123 Jan 18, 2020
62c30f6
multiple template parameter
jjsjann123 Feb 4, 2020
8e32aa2
Add some dependency tracking to Fusion.
csarofeen Feb 5, 2020
c784da7
Add ordering of exprs in Fuser. Returns list of exprs based on a topo…
csarofeen Feb 5, 2020
f3693f3
Maybe fix topo sort in function exprs in Fusion.h
csarofeen Feb 6, 2020
64a102e
fixing registration of val/expr to fusion and name_ assignment
jjsjann123 Feb 7, 2020
bf1d5fa
added tests; fix registration bug
jjsjann123 Feb 7, 2020
9ac00e7
Fix topo sort. Update topo sort test. Update printing int/float.
csarofeen Feb 7, 2020
8447262
adding failing test for topo sort
jjsjann123 Feb 7, 2020
c410060
Fix topo sort for intermediates being outputs.
csarofeen Feb 7, 2020
b54bcca
Put size checks back in test, test reverse addOutput order for topo s…
csarofeen Feb 7, 2020
f09daee
Re-write visitor as an iterative visitor that is topologically ordere…
csarofeen Feb 9, 2020
0a8ea53
Missed some cleanup when removing visitor.
csarofeen Feb 10, 2020
710826b
Simple fix for mutator test.
csarofeen Feb 10, 2020
81ede7e
Remove some duplicate dispatch.
csarofeen Feb 10, 2020
0aa5ecf
map torch::jit::fuser::Tensor directly to torch::jit::Value
jjsjann123 Feb 11, 2020
00b75de
Fix Val Type system, now split in ValType/DataType. Move Jie's Tensor…
csarofeen Feb 11, 2020
4f839b4
Comments around tensor, minor cleanup.
csarofeen Feb 11, 2020
9b47cf6
Create BinaryOp and UnaryOp Exprs.
kevinstephano Feb 11, 2020
6ffa095
Tensor class attempt 2.
jjsjann123 Feb 12, 2020
d2db595
renamed tensor tests
jjsjann123 Feb 12, 2020
6947c1e
Fix Val::getOrigin(...)
csarofeen Feb 12, 2020
4e74228
Remove stringify in type printing, add ParallelType.
csarofeen Feb 12, 2020
f274d86
Fix redeclare operator<<(... BinaryOpType...)
csarofeen Feb 12, 2020
b616416
[WIP] TensorDomain, TensorView, IterDomain.
csarofeen Feb 12, 2020
9bb395b
Add cast_op container function
kevinstephano Feb 12, 2020
0cc1424
Const in Arith. Test TensorView print.
csarofeen Feb 13, 2020
a458e62
Small fix and formating type.cpp
csarofeen Feb 13, 2020
d55767b
Add split, merge, reorder nodes.
csarofeen Feb 13, 2020
8fe8f85
TensorContiguity almost done
jjsjann123 Feb 13, 2020
0044afd
disabling failing tests. investigate later
jjsjann123 Feb 13, 2020
f58ea9b
Add convenience binary op alias e.g. Val* add(const Val*, const Val*)
csarofeen Feb 13, 2020
466f3f6
Int should take (int) as input, not float.
csarofeen Feb 13, 2020
6f1b785
Rename iriostream argument names to not conflict with lower case conv…
csarofeen Feb 13, 2020
8cced05
Add split function, format Tensor.cpp/.h
csarofeen Feb 13, 2020
38e540a
Actually push back the new domains in split.
csarofeen Feb 13, 2020
112872b
Add merge function.
csarofeen Feb 13, 2020
f8f69b7
fixing uninitialized domain pointer which gives undeterministic segfault
jjsjann123 Feb 13, 2020
ecfeddb
fixing test for fuser tensor
jjsjann123 Feb 13, 2020
13059ae
Implement split/merge for TensorView and Tensor. Tensor creates new T…
csarofeen Feb 14, 2020
9725e4e
Add reorder op.
csarofeen Feb 14, 2020
74f0ff5
Add some basic comparison for ir.h nodes.
csarofeen Feb 14, 2020
c9be830
Generate split/merge/reorder nodes in the IR, fix up printing, and ch…
csarofeen Feb 14, 2020
f8db677
Update dispatch for split/merge/reorder. Add comment on adding a Val/…
csarofeen Feb 15, 2020
6dfaa2b
Fill out same_as operators.
csarofeen Feb 15, 2020
6177e45
Update mutator functions.
csarofeen Feb 15, 2020
ec74f9e
Cleanup mutator, call removeExpr when setting a new origin Expr, and …
csarofeen Feb 16, 2020
3d12db2
Move to hierarchical dispatch. Add replaceAll mutator. Format ir.cppp.
csarofeen Feb 16, 2020
ce9f565
Add comment on dispatch.
csarofeen Feb 16, 2020
7e1669a
updated tests; minor changes on function returns
jjsjann123 Feb 16, 2020
2c18ffb
Debug spew cleanup.
csarofeen Feb 17, 2020
ea1f6e9
Add function Fusion::removeVal(..). Remove old val when replacing all…
csarofeen Feb 17, 2020
c6c73d9
finished TensorContiguity::merge
jjsjann123 Feb 18, 2020
c2aad3c
Move split/merge/reorder nodes to operate on Tensordomain. [WIP] Tran…
csarofeen Feb 18, 2020
72dcc37
Comment tensor.h describing the difference between Tensor, TensorView…
csarofeen Feb 18, 2020
d3d4504
Add inline operator string printing.
kevinstephano Feb 18, 2020
5fb80f1
Simple merge fix.
csarofeen Feb 19, 2020
2059792
Implement replay for compute_at operations. Add tests for it (print o…
csarofeen Feb 19, 2020
7d0e8b7
A couple bugs introduced during cleanup of replay for compute_at.
csarofeen Feb 19, 2020
a92a86c
Protect compute_at range from split/merge/reorder.
csarofeen Feb 19, 2020
5b97792
Quick fix calling tv->getComputeAtView() move reorder function before…
csarofeen Feb 19, 2020
63a8bce
Add traversal from specific values.
csarofeen Feb 19, 2020
bdc2497
Split/merge/reorder replaces tensor view in its origin operation.
csarofeen Feb 19, 2020
ce496f0
Add a utility to check if a value is in the dependency hierarchy of a…
csarofeen Feb 19, 2020
1996379
Add quick note on compute_at and it's axis/position.
csarofeen Feb 19, 2020
bfcb05d
Order of traversal was slightly off where lhs of an expression would …
csarofeen Feb 20, 2020
e8d8655
Replace all instances of view on split/merge/reoder, otherwise depend…
csarofeen Feb 20, 2020
a56a3c8
Fix reorder tests, as views now are hard replaced so can't be referen…
csarofeen Feb 20, 2020
7299295
Add dependency analysis, allow pulling dependencies between two values.
csarofeen Feb 21, 2020
8112547
Parser implemented that lowers JIT IR to codegen IR
jjsjann123 Feb 21, 2020
343fae9
Recrusive compute_at.
csarofeen Feb 21, 2020
3f821c4
Arith returns non const TensorViews.
csarofeen Feb 21, 2020
a4e08b8
Move transform replay impl to a .cpp file.
csarofeen Feb 21, 2020
aaf1bd8
Refactor const model. We want mutable members of val, so can't pass t…
csarofeen Feb 22, 2020
c1c5b9b
Refactor const model. We want mutable members of val, so can't pass t…
csarofeen Feb 22, 2020
8e59d7f
Make compute_at a member of TensorView.
csarofeen Feb 22, 2020
ade0338
Name val's by type.
csarofeen Feb 23, 2020
e7a500b
Some correctness fixes for transform replay.
csarofeen Feb 23, 2020
8cdf2a7
Propagate compute_at to other consumers of producer when necessary.
csarofeen Feb 24, 2020
2d3e625
Add TensorView to promote system.
csarofeen Feb 24, 2020
4b6fb98
Use arith functions on TensorView in tests.
csarofeen Feb 24, 2020
39e6365
Add breaking compute_at (reduction) example in comments in transform_…
csarofeen Feb 24, 2020
53a1ea4
Protect TensorView->tensor() as it can return nullptr.
csarofeen Feb 24, 2020
1aa3781
Properly support reduction axes in computeAt.
csarofeen Feb 24, 2020
53896f6
rewired FuseGraph to exclude concat/chunk while preserving chunk re-o…
jjsjann123 Feb 25, 2020
1d9e9be
adding a return for TensorView cout << operator
jjsjann123 Feb 25, 2020
be1261a
removing renamed include
jjsjann123 Feb 25, 2020
88057f0
computeAt directly modifies TensorView.
csarofeen Feb 26, 2020
7aae4ed
Add TwoAdds test and remove include from ir.h that breaks clean builds.
kevinstephano Feb 26, 2020
70421a9
Remove a spurious print test file.
kevinstephano Feb 26, 2020
84001f1
refactoring interface
jjsjann123 Feb 27, 2020
d5bc0cb
Add TensorView::clone, add TransformReplay::fullReplay, Minor cleanup.
csarofeen Feb 27, 2020
f8f5ace
Cleanup debug spew, minor fix to computeAt bounds check.
csarofeen Feb 27, 2020
8f02666
Reset TensorView stored compute_at_view before replay, otherwise coul…
csarofeen Feb 27, 2020
0840ff3
Add basic scaffolding for IR printing to start filling in. Added an …
kevinstephano Feb 27, 2020
db43978
Hooked Up IR Printing dispatching.
kevinstephano Feb 28, 2020
83a8886
refactored graph partition API
jjsjann123 Feb 28, 2020
cbecc14
Refactor pass logic out of transform_replay.
csarofeen Feb 28, 2020
5e4db46
Minor modifications to iter_visitor, move val types to last arg with …
csarofeen Feb 28, 2020
cca0f26
Hot fix in transform iter, it never called runBackward.
csarofeen Feb 28, 2020
ba2a730
[WIP] Basic code writing.
csarofeen Feb 28, 2020
bd2d1f1
Add predicates to printing, minor printing refactor, add LessThan bin…
csarofeen Mar 1, 2020
9826b0d
Making a parent class IRPrinter with two child classes, 1 for Operato…
kevinstephano Mar 2, 2020
a119de3
Minor fix, was creating an extra value in arith type promotion.
csarofeen Mar 2, 2020
fea3721
New test for CodeWrite. Some minor CodeWrite clean up.
csarofeen Mar 2, 2020
796d8f9
Rename printers to Math and Transform.
csarofeen Mar 2, 2020
e7f27ba
Fix IROperExprPrinter name change to IRMathPrinter
kevinstephano Mar 2, 2020
31ea7e1
Remove usage of dispatch, route through handle.
csarofeen Mar 2, 2020
147361b
Started populating Tensor Expressions.
kevinstephano Mar 3, 2020
39565e1
Disable CodeWrite, update iriostream to be based off a Printer class.
csarofeen Mar 3, 2020
3818cfe
Re-enable CodeWrite
csarofeen Mar 3, 2020
41e6ee6
Move predicate and index computation to separate headers.
csarofeen Mar 3, 2020
e4799a9
Change printer functions to virtual.
csarofeen Mar 3, 2020
7925579
Change CodeWrite to inherit from Printer.
csarofeen Mar 3, 2020
7738e3a
kernel cache added.
jjsjann123 Mar 3, 2020
3ac8540
Fixes to CodeWrite and Printer.
csarofeen Mar 3, 2020
fe7c763
dummy python code working for kernel caching and lowering
jjsjann123 Mar 4, 2020
4954449
Adding ForLoop Expression.
kevinstephano Mar 4, 2020
9be3d40
disabled move constructor for fusion;
jjsjann123 Mar 4, 2020
11f040d
Add assert in IterVisitor to ensuare traversal doesn't cross fusion b…
jjsjann123 Mar 4, 2020
a6e0bf7
Set fusion in FusionGuard in IterVisitor::Traverse as fusion* is an i…
csarofeen Mar 4, 2020
17414fb
[WIP] Code Write, printing thread bindings, for loops, predicates. Ne…
csarofeen Mar 4, 2020
6f12333
updated output allocation in kernel execution
jjsjann123 Mar 4, 2020
161bc16
Implement IfThenElse Expr in the IR.
kevinstephano Mar 4, 2020
fb046ed
pulled compilation into a separate file/module
jjsjann123 Mar 5, 2020
96fecc7
Flatten indexing into a tensor.
csarofeen Mar 5, 2020
82b7014
Cleanup IRPrinters.
csarofeen Mar 5, 2020
9f89990
[WIP] Codegen allocations.
csarofeen Mar 5, 2020
a2450ca
Fix indexing math into registers, some other minor fixes.
csarofeen Mar 6, 2020
d441736
Added TensorIndex Val class.
kevinstephano Mar 6, 2020
3a2a6b4
Bug fixes in CodeWrite.
csarofeen Mar 6, 2020
97fca1f
Refactor dispatch. Disable mutators, they still need to be refactored.
csarofeen Mar 6, 2020
0326c4e
Refactor dispatch system.
csarofeen Mar 6, 2020
8787101
In Code Writer add ForLoop expr IR node usage with populating express…
kevinstephano Mar 7, 2020
cd72b1a
Remove backup of ir.cpp.
csarofeen Mar 7, 2020
6279e95
[WIP] added code snippet for nvrtc compilation
jjsjann123 Mar 7, 2020
c5497b1
Refactor errors and error messages.
csarofeen Mar 8, 2020
45ca091
1. assert active fusion in printer;
jjsjann123 Mar 8, 2020
15030c3
Communication PR for Christian's refactor on CodeWrite
jjsjann123 Mar 8, 2020
fe477bf
Remove a couple nullptr checks.
csarofeen Mar 8, 2020
fd446f4
Remove start from ForLoop, we don't support non-zero start positions …
csarofeen Mar 8, 2020
20ad480
Minor changes to ForLoop print. Tab won't work across expressions, we…
csarofeen Mar 8, 2020
0e26078
Minor code write cleanup.
csarofeen Mar 8, 2020
4887ddb
Fix to indexing, need to check direction of size/stride in Tensor obj…
csarofeen Mar 8, 2020
0595242
Missed a TV print in CodeWrite should be T.
csarofeen Mar 8, 2020
bb95b87
Comment code_write
csarofeen Mar 9, 2020
ab53c80
Format code write
csarofeen Mar 9, 2020
6be452d
ComputeAt/transform bug fixes.
csarofeen Mar 9, 2020
f7f157c
Parse to code gen kernel.
csarofeen Mar 9, 2020
47326ac
Add string generation checks that are disabled until a bug is fixed t…
kevinstephano Mar 10, 2020
736a721
[CODEGEN in pipeline working]
jjsjann123 Mar 10, 2020
1789323
fixing failed tests via updating reference kernel
jjsjann123 Mar 10, 2020
51c7b68
Remove printing fusion in parser.cpp
csarofeen Mar 10, 2020
d868870
Move parser to arith functions to propagate sizes from inputs to outp…
csarofeen Mar 10, 2020
cbcd469
Move arith functions to camel-case.
csarofeen Mar 10, 2020
2ae9ec4
Don't let non-input/outputs participate in override in code_write
csarofeen Mar 10, 2020
01d00f1
removing HACK WAR as codegen works properly with intermediate storage
jjsjann123 Mar 10, 2020
a1c4c5d
Update CodeGen tests with Reference generated code matching.
kevinstephano Mar 10, 2020
555bc06
2f is not a valid float literal.
csarofeen Mar 10, 2020
67785fa
add break in switch statement, avoid false check failure on Scalar in…
jjsjann123 Mar 10, 2020
fd10130
[Parser update]
jjsjann123 Mar 10, 2020
4473198
Prepare for rebase to upstream/master: rm -rf torch/csrc/jit/compiler
zasdfgbnm Mar 10, 2020
31577b2
Prepare for rebase to upstream/master: rm -rf torch/csrc/jit/fuser/cu…
zasdfgbnm Mar 11, 2020
b9e3830
Prepare for rebase to upstream/master: replace some files with upstre…
zasdfgbnm Mar 11, 2020
8d0c769
Prepare for rebase to upstream/master: checkout jit/codegen from upst…
zasdfgbnm Mar 11, 2020
82e3b7d
Prepare for rebase to upstream/master: move to codegen/cuda
zasdfgbnm Mar 11, 2020
785cb76
fix build
zasdfgbnm Mar 11, 2020
7aafdf8
remove copied files
zasdfgbnm Mar 11, 2020
c30a992
fix compile error
zasdfgbnm Mar 11, 2020
e4bb5e7
fix linking error
zasdfgbnm Mar 11, 2020
3c9cb67
remove operator list from partition.cpp
zasdfgbnm Mar 11, 2020
0e6da13
revert changes in test/cpp/jit/test_misc.cpp
zasdfgbnm Mar 11, 2020
4e19b78
Hotfix for new val construction during arith ops.
csarofeen Mar 11, 2020
c6b60b3
Code cleanup and commenting of arith.cpp/h
csarofeen Mar 11, 2020
1a8ad0d
Code cleanup and commenting of fusion.cpp/h
csarofeen Mar 11, 2020
0085ea7
Minor tweak to test.
csarofeen Mar 11, 2020
6ba73ca
Minor cleanup to code_write.h
csarofeen Mar 11, 2020
c05c7b6
Format and comment dispatch.h
csarofeen Mar 11, 2020
d075bde
Include dispatch in iriostream.
csarofeen Mar 11, 2020
36fea51
Minor cleanup to iter_visitor.cpp .
csarofeen Mar 11, 2020
916d6af
add back cuda fusion pass lost in rebasing
jjsjann123 Mar 11, 2020
263b6d8
Fix CodeGen tests by adding an insertion order based stucture, a vect…
kevinstephano Mar 11, 2020
71935ad
Comment index/predicate compute. Factor into .h and .cpp files.
csarofeen Mar 11, 2020
7647ef7
Split IR nodes into ir_base, ir_nodes, tensor.
csarofeen Mar 12, 2020
72526e0
Add Test ExecKernel that manually passes a generated kernel to test v…
kevinstephano Mar 12, 2020
00f3ffd
replaced prim::FusionGroup with prim::CudaFusionGroup in cuda codegen
jjsjann123 Mar 11, 2020
501ced1
refactored parser & partition
jjsjann123 Mar 12, 2020
65f8f6d
update build files
jjsjann123 Mar 12, 2020
1540bbe
graph partition pass is not correct. But at least build issue is gone…
jjsjann123 Mar 12, 2020
f7e05ab
Another pass at commenting ir_base
csarofeen Mar 12, 2020
f041286
Cleanup/comment ir_nodes. Change same_as to sameAs.
csarofeen Mar 12, 2020
e41ff76
Quick interface header to get all needed items for codegen.
csarofeen Mar 12, 2020
7aa9262
Short comment for iriostream.
csarofeen Mar 12, 2020
038303c
Refactor TensorView, add comment, add member functions split, merge, …
csarofeen Mar 12, 2020
d8243a1
Avoid TensorView->domain() calls when possible.
csarofeen Mar 12, 2020
f46f87b
fixing parser operator registration
jjsjann123 Mar 12, 2020
a57b238
Break nodes up into 3 headers, ir_base_nodes, ir_internal_nodes, ir_i…
csarofeen Mar 12, 2020
68ea130
Remove old header file from being included.
csarofeen Mar 12, 2020
0c09f9e
[pipeline work cleared] Pipeline working again!
jjsjann123 Mar 12, 2020
577e1b8
code cleaning.
jjsjann123 Mar 13, 2020
142feb3
Modified the SimplePWise test to compile and execute its code. Also …
kevinstephano Mar 13, 2020
09728ea
Updated CodeGen2 Test such that computeAt is on the inner most dimens…
kevinstephano Mar 13, 2020
140c20c
removed TensorContiguity from Tensor, in preparation to remove Tensor
jjsjann123 Mar 13, 2020
53402b1
Remove Tensor::MakeDummyTensor(...), remove some tests.
csarofeen Mar 13, 2020
48379e1
Remove Tensor from codegen IR.
csarofeen Mar 13, 2020
4be2007
Fix tests after removing Tensor from codegen IR.
csarofeen Mar 13, 2020
f3b5d61
Remove debug spew, format kernel.h
csarofeen Mar 13, 2020
e3a60a3
Refactor printing in tests.
csarofeen Mar 13, 2020
510a44c
Refactor merge test, remove computeAt tests.
csarofeen Mar 13, 2020
0c37004
fixing parser initialization
jjsjann123 Mar 13, 2020
6025f20
uncomment the parser test
jjsjann123 Mar 13, 2020
2e55326
removed obsolete files; added comments
jjsjann123 Mar 13, 2020
802e79d
python test added
jjsjann123 Mar 14, 2020
ec2717a
Rebase in process.
csarofeen Mar 14, 2020
e29e00e
Remove old files.
csarofeen Mar 14, 2020
61423f0
Revert files we didn't want to touch.
csarofeen Mar 14, 2020
12e90b6
More cleanup back to master.
csarofeen Mar 14, 2020
6b92908
In progress rebase.
csarofeen Mar 15, 2020
2f1dcdc
Finish up rebase, remove last bit of printing in test_gpu_fusion.cpp
csarofeen Mar 15, 2020
09dce6a
Minor changes in python tests for new cuda fuser.
csarofeen Mar 15, 2020
aebb5aa
Remove ForLoop, TensorIndex, and IfThenElse nodes for initial PR.
csarofeen Mar 15, 2020
6d78a01
Rename {Pre,Post}FusionPass -> {Pre,Post}Pass.
csarofeen Mar 15, 2020
fd33f60
Fusible->Fusable
csarofeen Mar 15, 2020
624681b
fixing cpu only builds
jjsjann123 Mar 16, 2020
File filter

Filter by extension

Filter by extension


Conversations
Failed to load comments.
Loading
Jump to
The table of contents is too big for display.
Diff view
Diff view
  •  
  •  
  •  
4 changes: 2 additions & 2 deletions .circleci/cimodel/data/binary_build_definitions.py
Original file line number Diff line number Diff line change
Expand Up @@ -33,8 +33,8 @@ def gen_docker_image(self):

docker_distro_prefix = miniutils.override(self.pydistro, docker_word_substitution)

# The cpu nightlies are built on the pytorch/manylinux-cuda100 docker image
alt_docker_suffix = self.cuda_version or "100"
# The cpu nightlies are built on the pytorch/manylinux-cuda102 docker image
alt_docker_suffix = self.cuda_version or "102"
docker_distro_suffix = "" if self.pydistro == "conda" else alt_docker_suffix
return miniutils.quote("pytorch/" + docker_distro_prefix + "-cuda" + docker_distro_suffix)

Expand Down
1 change: 0 additions & 1 deletion .circleci/cimodel/data/dimensions.py
Original file line number Diff line number Diff line change
Expand Up @@ -3,7 +3,6 @@
CUDA_VERSIONS = [
None, # cpu build
"92",
"100",
"101",
"102",
]
Expand Down
1 change: 0 additions & 1 deletion .circleci/cimodel/data/pytorch_build_data.py
Original file line number Diff line number Diff line change
Expand Up @@ -42,7 +42,6 @@
X("3.6"),
]),
("9.2", [X("3.6")]),
("10", [X("3.6")]),
("10.1", [X("3.6")]),
("10.2", [
XImportant("3.6"),
Expand Down
Loading